|
|
@@ -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
|
|
|
|
|
|
|