yjh0410 2 éve
szülő
commit
5ea53f89a1
1 módosított fájl, 3 hozzáadás és 1 törlés
  1. 3 1
      config/__init__.py

+ 3 - 1
config/__init__.py

@@ -44,7 +44,7 @@ from .data_config.transform_config import (
 def build_trans_config(trans_config='ssd'):
     print('==============================')
     print('Transform: {}-Style ...'.format(trans_config))
-    
+   
     # SSD-style transform 
     if trans_config == 'ssd':
         cfg = ssd_trans_config
@@ -91,6 +91,8 @@ def build_trans_config(trans_config='ssd'):
     elif trans_config == 'yolovx_huge':
         cfg = yolovx_huge_trans_config
 
+    print('Transform Config: {} \n'.format(cfg))
+
     return cfg