From 7ad69cc3f6041fc5c59ec522e5e3c20c3cf692a8 Mon Sep 17 00:00:00 2001
From: wangguanzhong <jerrywgz@126.com>
Date: Fri, 1 Nov 2019 15:32:57 +0800
Subject: [PATCH] add info for obj365 download (#12)

* add info for obj365 download
---
 ppdet/utils/download.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/ppdet/utils/download.py b/ppdet/utils/download.py
index ee8a0eaed..00c3ec8c5 100644
--- a/ppdet/utils/download.py
+++ b/ppdet/utils/download.py
@@ -74,6 +74,7 @@ DATASETS = {
     'fruit': ([(
         'https://dataset.bj.bcebos.com/PaddleDetection_demo/fruit-detection.tar',
         'ee4a1bf2e321b75b0850cc6e063f79d7', ), ], ["fruit-detection"]),
+    'objects365': (),
 }
 
 DOWNLOAD_RETRY_LIMIT = 3
@@ -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
-- 
GitLab