소스 검색

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