Skip to content

Commit

Permalink
add info for obj365 download (PaddlePaddle#12)
Browse files Browse the repository at this point in the history
* add info for obj365 download
  • Loading branch information
jerrywgz authored Nov 1, 2019
1 parent f10b52e commit 7ad69cc
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion ppdet/utils/download.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,7 @@
'fruit': ([(
'https://dataset.bj.bcebos.com/PaddleDetection_demo/fruit-detection.tar',
'ee4a1bf2e321b75b0850cc6e063f79d7', ), ], ["fruit-detection"]),
'objects365': (),
}

DOWNLOAD_RETRY_LIMIT = 3
Expand Down Expand Up @@ -104,8 +105,11 @@ def get_dataset_path(path, annotation, image_dir):
if os.path.split(path.strip().lower())[-1] == name:
logger.info("Parse dataset_dir {} as dataset "
"{}".format(path, name))
if name == 'objects365':
raise NotImplementedError(
"Dataset {} is not valid for download automatically. Please apply and download the dataset from https://www.objects365.org/download.html".
format(name))
data_dir = osp.join(DATASET_HOME, name)

# For voc, only check dir VOCdevkit/VOC2012, VOCdevkit/VOC2007
if name == 'voc':
exists = True
Expand Down

0 comments on commit 7ad69cc

Please sign in to comment.