Detection on the test set done!
recoded the hardcoding of image sizes for detection