update

yolo/config/yolov2_config.py

@@ -28,7 +28,7 @@ class Yolov2BaseConfig(object):
         self.val_conf_thresh = 0.001
         self.val_nms_thresh  = 0.7
         self.test_topk = 300
-        self.test_conf_thresh = 0.3
+        self.test_conf_thresh = 0.4
         self.test_nms_thresh  = 0.5
 
         # ---------------- Assignment config ----------------

yolo/config/yolov3_config.py

@@ -26,7 +26,7 @@ class Yolov3Config(object):
         self.val_conf_thresh = 0.001
         self.val_nms_thresh  = 0.7
         self.test_topk = 100
-        self.test_conf_thresh = 0.45
+        self.test_conf_thresh = 0.4
         self.test_nms_thresh  = 0.5
 
         # ---------------- Assignment config ----------------

yolo/models/yolov2/yolov2.py

@@ -30,7 +30,7 @@ class Yolov2(nn.Module):
         
         # ---------------------- Network Parameters ----------------------
         self.backbone = Yolov2Backbone(cfg)
-        self.neck     = SPPF(cfg, self.backbone.feat_dim, cfg.head_dim)
+        self.neck     = SPPF(self.backbone.feat_dim, cfg.head_dim)
         self.head     = Yolov2DetHead(cfg, self.neck.out_dim)
         self.pred     = Yolov2DetPredLayer(cfg)