format

2afb5a2f · yhcao6 · 8c94b2d8 · 2afb5a2f · 2afb5a2f
Commit 2afb5a2f authored Dec 06, 2018 by yhcao6
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

mmdet/datasets/__init__.py mmdet/datasets/__init__.py +0 -1

mmdet/datasets/utils.py mmdet/datasets/utils.py +2 -1

No files found.
--- a/mmdet/datasets/__init__.py
+++ b/mmdet/datasets/__init__.py
@@ -10,4 +10,3 @@ __all__ = [
    'build_dataloader', 'to_tensor', 'random_scale', 'show_ann',
    'get_dataset', 'ExtraAugmentation', 'ConcatDataset', 'RepeatDataset',
 ]
-
--- a/mmdet/datasets/utils.py
+++ b/mmdet/datasets/utils.py
@@ -76,7 +76,8 @@ def show_ann(coco, img, ann_info):

 def get_dataset(data_cfg):
    if data_cfg['type'] == 'RepeatDataset':
-        return RepeatDataset(get_dataset(data_cfg['dataset']), data_cfg['times'])
+        return RepeatDataset(
+            get_dataset(data_cfg['dataset']), data_cfg['times'])

    if isinstance(data_cfg['ann_file'], (list, tuple)):
        ann_files = data_cfg['ann_file']