Skip to content
Snippets Groups Projects
Commit 2afb5a2f authored by yhcao6's avatar yhcao6
Browse files

format

parent 8c94b2d8
No related branches found
No related tags found
No related merge requests found
...@@ -10,4 +10,3 @@ __all__ = [ ...@@ -10,4 +10,3 @@ __all__ = [
'build_dataloader', 'to_tensor', 'random_scale', 'show_ann', 'build_dataloader', 'to_tensor', 'random_scale', 'show_ann',
'get_dataset', 'ExtraAugmentation', 'ConcatDataset', 'RepeatDataset', 'get_dataset', 'ExtraAugmentation', 'ConcatDataset', 'RepeatDataset',
] ]
...@@ -76,7 +76,8 @@ def show_ann(coco, img, ann_info): ...@@ -76,7 +76,8 @@ def show_ann(coco, img, ann_info):
def get_dataset(data_cfg): def get_dataset(data_cfg):
if data_cfg['type'] == 'RepeatDataset': 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)): if isinstance(data_cfg['ann_file'], (list, tuple)):
ann_files = data_cfg['ann_file'] ann_files = data_cfg['ann_file']
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment