From 2afb5a2faff48d9cc127732b37e25044f02ef641 Mon Sep 17 00:00:00 2001 From: yhcao6 <yhcao6@gmail.com> Date: Thu, 6 Dec 2018 23:21:37 +0800 Subject: [PATCH] format --- mmdet/datasets/__init__.py | 1 - mmdet/datasets/utils.py | 3 ++- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mmdet/datasets/__init__.py b/mmdet/datasets/__init__.py index 9c29482..2e72022 100644 --- 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', ] - diff --git a/mmdet/datasets/utils.py b/mmdet/datasets/utils.py index b9ee9a8..8fdba7f 100644 --- 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'] -- GitLab