update config files

SangbumChoi · Apr 24, 2023 · 8b941c0 · 8b941c0
1 parent 6b9f5f4
commit 8b941c0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/configs/base/yolov6l_base_finetune.py b/configs/base/yolov6l_base_finetune.py
@@ -9,9 +9,10 @@
         num_repeats=[1, 6, 12, 18, 6],
         out_channels=[64, 128, 256, 512, 1024],
         csp_e=float(1)/2,
+        fuse_P2=True,
         ),
     neck=dict(
-        type='CSPRepPANNeck',
+        type='CSPRepBiFPANNeck',
         num_repeats=[12, 12, 12, 12],
         out_channels=[256, 128, 128, 256, 256, 512],
         csp_e=float(1)/2,

diff --git a/configs/base/yolov6s_base_finetune.py b/configs/base/yolov6s_base_finetune.py
@@ -13,7 +13,7 @@
         cspsppf=True,
         ),
     neck=dict(
-        type='CSPRepPANNeck',
+        type='CSPRepBiFPANNeck',
         num_repeats=[12, 12, 12, 12],
         out_channels=[256, 128, 128, 256, 256, 512],
         csp_e=float(1)/2,