|
|
@@ -41,7 +41,7 @@ yolovx_cfg = {
|
|
|
# ---------------- Train config ----------------
|
|
|
## Input
|
|
|
'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
- 'trans_type': 'yolox_pico',
|
|
|
+ 'trans_type': 'rtmdet_pico',
|
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
|
'matcher': {'center_sampling_radius': 2.5,
|
|
|
@@ -95,7 +95,7 @@ yolovx_cfg = {
|
|
|
# ---------------- Train config ----------------
|
|
|
## Input
|
|
|
'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
- 'trans_type': 'yolox_nano',
|
|
|
+ 'trans_type': 'rtmdet_nano',
|
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
|
'matcher': {'center_sampling_radius': 2.5,
|
|
|
@@ -149,7 +149,7 @@ yolovx_cfg = {
|
|
|
# ---------------- Train config ----------------
|
|
|
## Input
|
|
|
'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
- 'trans_type': 'yolox_nano',
|
|
|
+ 'trans_type': 'rtmdet_nano',
|
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
|
'matcher': {'center_sampling_radius': 2.5,
|
|
|
@@ -203,7 +203,7 @@ yolovx_cfg = {
|
|
|
# ---------------- Train config ----------------
|
|
|
## Input
|
|
|
'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
- 'trans_type': 'yolox_small',
|
|
|
+ 'trans_type': 'rtmdet_small',
|
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
|
'matcher': {'center_sampling_radius': 2.5,
|
|
|
@@ -257,7 +257,7 @@ yolovx_cfg = {
|
|
|
# ---------------- Train config ----------------
|
|
|
## Input
|
|
|
'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
- 'trans_type': 'yolox_medium',
|
|
|
+ 'trans_type': 'rtmdet_medium',
|
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
|
'matcher': {'center_sampling_radius': 2.5,
|
|
|
@@ -311,7 +311,7 @@ yolovx_cfg = {
|
|
|
# ---------------- Train config ----------------
|
|
|
## Input
|
|
|
'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
- 'trans_type': 'yolox_large',
|
|
|
+ 'trans_type': 'rtmdet_large',
|
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
|
'matcher': {'center_sampling_radius': 2.5,
|
|
|
@@ -365,7 +365,7 @@ yolovx_cfg = {
|
|
|
# ---------------- Train config ----------------
|
|
|
## Input
|
|
|
'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
- 'trans_type': 'yolox_huge',
|
|
|
+ 'trans_type': 'rtmdet_huge',
|
|
|
# ---------------- Assignment config ----------------
|
|
|
## Matcher
|
|
|
'matcher': {'center_sampling_radius': 2.5,
|
yjh0410