Commit 8c94b2d8 authored by yhcao6's avatar yhcao6
Browse files

fix

parent 63288def
......@@ -76,7 +76,7 @@ def show_ann(coco, img, ann_info):
def get_dataset(data_cfg):
if data_cfg['type'] == 'RepeatDataset':
return RepeatDataset(get_dataset(data_cfg['type']), 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']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment