浏览代码

debug RTRDet-L on COCO

yjh0410 2 年之前
父节点
当前提交
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)