yjh0410 2 năm trước cách đây
mục cha
commit
a9bc32dbf8

+ 1 - 1
config/model_config/yolovx_config.py

@@ -311,7 +311,7 @@ yolovx_cfg = {
         # ---------------- Train config ----------------
         # ---------------- Train config ----------------
         ## Input
         ## Input
         'multi_scale': [0.5, 1.25],   # 320 -> 960
         'multi_scale': [0.5, 1.25],   # 320 -> 960
-        'trans_type': 'rtmdet_large',
+        'trans_type': 'yolox_large',
         # ---------------- Assignment config ----------------
         # ---------------- Assignment config ----------------
         ## Matcher
         ## Matcher
         'matcher': {'center_sampling_radius': 2.5,
         'matcher': {'center_sampling_radius': 2.5,