yjh0410 2 лет назад
Родитель
Сommit
e26d856c1b
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      engine.py

+ 1 - 1
engine.py

@@ -1287,7 +1287,7 @@ class RTRTrainer(object):
             # Multi scale
             if self.args.multi_scale:
                 images, targets, img_size = self.rescale_image_targets(
-                    images, targets, self.model_cfg['max_stride'], self.args.min_box_size, self.model_cfg['multi_scale'])
+                    images, targets, 32, self.args.min_box_size, self.model_cfg['multi_scale'])
             else:
                 targets = self.refine_targets(targets, self.args.min_box_size)