yjh0410 1 år sedan
förälder
incheckning
fb77152c7e
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      odlab/config/yolof_config.py

+ 2 - 2
odlab/config/yolof_config.py

@@ -89,8 +89,8 @@ class YolofBaseConfig(object):
         ## input size
         self.train_min_size = [800]   # short edge of image
         self.train_max_size = 1333
-        self.test_min_size  = [800]
-        self.test_max_size  = 1333
+        self.test_min_size  = [320]
+        self.test_max_size  = 320
         ## Pixel mean & std
         self.pixel_mean = [0.485, 0.456, 0.406]
         self.pixel_std  = [0.229, 0.224, 0.225]