Add check_requirements(('pycocotools>=2.0',)) (#7295)

Add `check_requirements(('pycocotools>=2.0',))`
ultralytics · Apr 5, 2022 · c759bbd · c759bbd
1 parent b1300f3
commit c759bbd
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/data/Objects365.yaml b/data/Objects365.yaml
@@ -60,11 +60,12 @@ names: ['Person', 'Sneakers', 'Chair', 'Other Shoes', 'Hat', 'Car', 'Lamp', 'Gla
 
 # Download script/URL (optional) ---------------------------------------------------------------------------------------
 download: |
-  from pycocotools.coco import COCO
   from tqdm import tqdm
-
-  from utils.general import Path, download, np, xyxy2xywhn
-
+  
+  from utils.general import Path, check_requirements, download, np, xyxy2xywhn
+  
+  check_requirements(('pycocotools>=2.0',))
+  from pycocotools.coco import COCO
 
   # Make Directories
   dir = Path(yaml['path'])  # dataset root dir