chore(examples): update import in exps(#42)

chore(examples): update import in exps
Megvii-BaseDetection · Aug 6, 2021 · 38e3c56 · 38e3c56
1 parent b861b22
commit 38e3c56
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/exps/default/yolov3.py b/exps/default/yolov3.py
@@ -34,12 +34,17 @@ def init_yolo(M):
         return self.model
 
     def get_data_loader(self, batch_size, is_distributed, no_aug=False):
-        from data.datasets.cocodataset import COCODataset
-        from data.datasets.mosaicdetection import MosaicDetection
-        from data.datasets.data_augment import TrainTransform
-        from data.datasets.dataloading import YoloBatchSampler, DataLoader, InfiniteSampler
         import torch.distributed as dist
 
+        from yolox.data import (
+            COCODataset,
+            DataLoader,
+            InfiniteSampler,
+            MosaicDetection,
+            TrainTransform,
+            YoloBatchSampler
+        )
+
         dataset = COCODataset(
                 data_dir='data/COCO/',
                 json_file=self.train_ann,