yjh0410 2 anni fa
parent
commit
b16efd03e2

+ 4 - 4
config/transform_config.py

@@ -77,7 +77,7 @@ yolov5_small_trans_config = {
     'mixup_scale': [0.5, 1.5]   # "mixup_scale" is not used for YOLOv5MixUp
 }
 
-yolov5_tiny_trans_config = {
+yolov5_nano_trans_config = {
     'aug_type': 'yolov5',
     # Basic Augment
     'degrees': 0.0,
@@ -96,7 +96,7 @@ yolov5_tiny_trans_config = {
     'mixup_scale': [0.5, 1.5]   # "mixup_scale" is not used for YOLOv5MixUp
 }
 
-yolov5_nano_trans_config = {
+yolov5_pico_trans_config = {
     'aug_type': 'yolov5',
     # Basic Augment
     'degrees': 0.0,
@@ -192,7 +192,7 @@ yolox_small_trans_config = {
     'mixup_scale': [0.5, 1.5]   # "mixup_scale" is not used for YOLOv5MixUp
 }
 
-yolox_tiny_trans_config = {
+yolox_nano_trans_config = {
     'aug_type': 'yolov5',
     # Basic Augment
     'degrees': 0.0,
@@ -211,7 +211,7 @@ yolox_tiny_trans_config = {
     'mixup_scale': [0.5, 1.5]   # "mixup_scale" is not used for YOLOv5MixUp
 }
 
-yolox_nano_trans_config = {
+yolox_pico_trans_config = {
     'aug_type': 'yolov5',
     # Basic Augment
     'degrees': 0.0,

+ 1 - 1
config/yolov3_config.py

@@ -96,7 +96,7 @@ yolov3_cfg = {
                         [116, 90],  [156, 198], [373, 326]],  # P5
         # ---------------- Train config ----------------
         ## input
-        'trans_type': 'yolov5_tiny',
+        'trans_type': 'yolov5_nano',
         'multi_scale': [0.5, 1.0],
         # ---------------- Assignment config ----------------
         ## matcher

+ 1 - 1
config/yolov4_config.py

@@ -96,7 +96,7 @@ yolov4_cfg = {
                         [116, 90],  [156, 198], [373, 326]],  # P5
         # ---------------- Train config ----------------
         ## input
-        'trans_type': 'yolov5_tiny',
+        'trans_type': 'yolov5_nano',
         'multi_scale': [0.5, 1.0],
         # ---------------- Assignment config ----------------
         ## matcher

+ 1 - 1
config/yolov5_config.py

@@ -34,7 +34,7 @@ yolov5_cfg = {
         # ---------------- Train config ----------------
         ## input
         'multi_scale': [0.5, 1.0],   # 320 -> 640
-        'trans_type': 'yolov5_tiny',
+        'trans_type': 'yolov5_nano',
         # ---------------- Assignment config ----------------
         ## matcher
         'anchor_thresh': 4.0,

+ 1 - 1
config/yolov7_config.py

@@ -3,7 +3,7 @@
 yolov7_cfg = {
     'yolov7_t':{
         # input
-        'trans_type': 'yolov5_tiny',
+        'trans_type': 'yolov5_nano',
         'multi_scale': [0.5, 1.5], # 320 -> 960
         # model
         'backbone': 'elannet_tiny',

+ 1 - 1
config/yolox_config.py

@@ -32,7 +32,7 @@ yolox_cfg = {
         # ---------------- Train config ----------------
         ## input
         'multi_scale': [0.5, 1.5],   # 320 -> 960
-        'trans_type': 'yolox_tiny',
+        'trans_type': 'yolox_nano',
         # ---------------- Assignment config ----------------
         ## matcher
         'matcher': {'center_sampling_radius': 2.5,