Преглед на файлове

use 320-800 multi scales for CTRNet

yjh0410 преди 1 година
родител
ревизия
76238ff308
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      config/model_config/ctrnet_config.py

+ 1 - 1
config/model_config/ctrnet_config.py

@@ -36,7 +36,7 @@ ctrnet_cfg = {
         'head_depthwise': False,  
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.0],   # 320 -> 640
+        'multi_scale': [0.5, 1.25],   # 320 -> 800
         'trans_type': 'yolov5_s',
         # ---------------- Assignment config ----------------
         ## Matcher