|
@@ -43,7 +43,7 @@ rtcdet_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_n',
|
|
'trans_type': 'yolov5_n',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
|
@@ -103,7 +103,7 @@ rtcdet_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_s',
|
|
'trans_type': 'yolov5_s',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
|
@@ -163,7 +163,7 @@ rtcdet_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_m',
|
|
'trans_type': 'yolov5_m',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
|
@@ -223,7 +223,7 @@ rtcdet_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_l',
|
|
'trans_type': 'yolov5_l',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
|
@@ -283,7 +283,7 @@ rtcdet_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_x',
|
|
'trans_type': 'yolov5_x',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
|
@@ -348,7 +348,7 @@ rtcdet_seg_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_n',
|
|
'trans_type': 'yolov5_n',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
|
@@ -412,7 +412,7 @@ rtcdet_pos_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_n',
|
|
'trans_type': 'yolov5_n',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
|
@@ -476,7 +476,7 @@ rtcdet_seg_pos_cfg = {
|
|
|
},
|
|
},
|
|
|
# ---------------- Train config ----------------
|
|
# ---------------- Train config ----------------
|
|
|
## input
|
|
## input
|
|
|
- 'multi_scale': [0.5, 1.5], # 320 -> 960
|
|
|
|
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
'trans_type': 'yolov5_n',
|
|
'trans_type': 'yolov5_n',
|
|
|
# ---------------- Assignment config ----------------
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
## Matcher
|
yjh0410