Fix issues: Class Name @BDeMo

Author: BDeMo

dd_ranking/__init__.py

@@ -1,5 +1,5 @@
-from .aug import *
-from .config import *
-from .loss import *
-from .metrics import *
-from .utils import *
+import dd_ranking.aug as aug
+import dd_ranking.config as config
+import dd_ranking.loss as loss
+import dd_ranking.metrics as metrics
+import dd_ranking.utils as utils

dd_ranking/metrics/hard_label.py

@@ -12,7 +12,7 @@
 from dd_ranking.utils import build_model, get_pretrained_model_path
 from dd_ranking.utils import TensorDataset, get_random_images, get_dataset, save_results
 from dd_ranking.utils import set_seed, train_one_epoch, validate, get_optimizer, get_lr_scheduler
-from dd_ranking.aug import DSA_Augmentation, ZCA_Whitening_Augmentation, Mixup_Augmentation, Cutmix_Augmentation
+from dd_ranking.aug import DSAugmentation, MixupAugmentation, CutmixAugmentation, ZCAWhiteningAugmentation
 from dd_ranking.config import Config
 
 

dd_ranking/metrics/soft_label.py

@@ -13,7 +13,7 @@
 from dd_ranking.utils import TensorDataset, get_random_images, get_dataset, save_results
 from dd_ranking.utils import set_seed, train_one_epoch, validate, get_optimizer, get_lr_scheduler
 from dd_ranking.loss import SoftCrossEntropyLoss, KLDivergenceLoss
-from dd_ranking.aug import DSA_Augmentation, ZCA_Whitening_Augmentation, Mixup_Augmentation, Cutmix_Augmentation
+from dd_ranking.aug import DSAugmentation, MixupAugmentation, CutmixAugmentation, ZCAWhiteningAugmentation
 from dd_ranking.config import Config