debug YOLOX

models/yolox/yolox.py

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

models/yolox/yolox_backbone.py

@@ -70,7 +70,7 @@ def build_backbone(cfg, pretrained=False):
     feat_dims = backbone.feat_dims
 
     if pretrained:
-        if cfg['width'] == 1.0 and  cfg['depth'] == 1.0:
+        if cfg['width'] == 1.0 and cfg['depth'] == 1.0:
             url = model_urls['cspdarknet_large']
         if url is not None:
             print('Loading pretrained weight ...')