debug YOLOX

models/yolox/yolox.py

@@ -31,7 +31,8 @@ class YOLOX(nn.Module):
         
         # ------------------- Network Structure -------------------
         ## 主干网络
-        self.backbone, feats_dim = build_backbone(cfg=cfg)
+        self.backbone, feats_dim = build_backbone(
+            cfg['backbone'], trainable&cfg['pretrained'])
         
         ## 颈部网络: 特征金字塔
         self.fpn = build_fpn(cfg=cfg, in_dims=feats_dim, out_dim=int(256*cfg['width']))

utils/box_ops.py

@@ -89,7 +89,6 @@ def rescale_bboxes(bboxes, origin_img_size, cur_img_size, deltas=None):
         # rescale
         bboxes[..., [0, 2]] = bboxes[..., [0, 2]] / (cur_img_w - deltas[0]) * origin_w
         bboxes[..., [1, 3]] = bboxes[..., [1, 3]] / (cur_img_h - deltas[1]) * origin_h
-        print('deltas')
         
         # clip bboxes
         bboxes[..., [0, 2]] = np.clip(bboxes[..., [0, 2]], a_min=0., a_max=origin_w)