浏览代码

modify rtcdet's matcher

yjh0410 1 年之前
父节点
当前提交
c965a28723
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      yolo/config/rtcdet_config.py

+ 1 - 1
yolo/config/rtcdet_config.py

@@ -214,7 +214,7 @@ class RTCDetXConfig(RTCDetBaseConfig):
         self.last_stage_ratio = 1.0
         self.num_blocks = [3, 6, 6, 3]
         self.scale = "x"
-        self.fpn_num_blocks = 4
+        self.fpn_num_blocks = 3
 
         # ---------------- Data process config ----------------
         self.mosaic_prob = 1.0