Jelajahi Sumber

remove YOLOX-style augment

yjh0410 2 tahun lalu
induk
melakukan
a8a3dcb8a3
1 mengubah file dengan 0 tambahan dan 9 penghapusan
  1. 0 9
      config/__init__.py

+ 0 - 9
config/__init__.py

@@ -43,7 +43,6 @@ def build_model_config(args):
 # ------------------ Transform Config ----------------------
 from .transform_config import (
     yolov5_strong_trans_config, yolov5_weak_trans_config, yolov5_nano_trans_config,
-    yolox_strong_trans_config, yolox_weak_trans_config, yolox_nano_trans_config,
     ssd_trans_config
 )
 
@@ -62,12 +61,4 @@ def build_trans_config(trans_config='ssd'):
     elif trans_config == 'yolov5_nano':
         cfg = yolov5_nano_trans_config
         
-    # YOLOX-style transform 
-    elif trans_config == 'yolox_strong':
-        cfg = yolox_strong_trans_config
-    elif trans_config == 'yolox_weak':
-        cfg = yolox_weak_trans_config
-    elif trans_config == 'yolox_nano':
-        cfg = yolox_nano_trans_config
-
     return cfg