Modify ROOT_PATH in gen_data.py

datasets/gen_data.py

@@ -213,7 +213,9 @@ parser.add_argument('--mode', default='all',
 opt = parser.parse_args()
 
 if __name__ == "__main__":
-    ROOT_PATH = './'
+    ROOT_PATH = os.getcwd()
+    if not os.path.exists(ROOT_PATH+"/labels"):
+        ROOT_PATH = os.path.join(ROOT_PATH, "datasets")
     mode_type = opt.mode
     if mode_type == 'all':
         deal_list = ['train', 'val']