New ASSETS and trackers GMC cleanup (#4425)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

ultralytics/data/converter.py

@@ -1,6 +1,7 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
 import json
+import shutil
 from collections import defaultdict
 from pathlib import Path
 
@@ -9,7 +10,6 @@ import numpy as np
 from tqdm import tqdm
 
 from ultralytics.utils.checks import check_requirements
-from ultralytics.utils.files import make_dirs
 
 
 def coco91_to_coco80_class():
@@ -27,6 +27,27 @@ def coco91_to_coco80_class():
         None, 73, 74, 75, 76, 77, 78, 79, None]
 
 
+def coco80_to_coco91_class():  #
+    """
+    Converts 80-index (val2014) to 91-index (paper).
+    For details see https://tech.amikelive.com/node-718/what-object-categories-labels-are-in-coco-dataset/.
+
+    Example:
+        ```python
+        import numpy as np
+
+        a = np.loadtxt('data/coco.names', dtype='str', delimiter='\n')
+        b = np.loadtxt('data/coco_paper.names', dtype='str', delimiter='\n')
+        x1 = [list(a[i] == b).index(True) + 1 for i in range(80)]  # darknet to coco
+        x2 = [list(b[i] == a).index(True) if any(b[i] == a) else None for i in range(91)]  # coco to darknet
+        ```
+    """
+    return [
+        1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 27, 28, 31, 32, 33, 34,
+        35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63,
+        64, 65, 67, 70, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 84, 85, 86, 87, 88, 89, 90]
+
+
 def convert_coco(labels_dir='../coco/annotations/', use_segments=False, use_keypoints=False, cls91to80=True):
     """Converts COCO dataset annotations to a format suitable for training YOLOv5 models.
 
@@ -47,7 +68,14 @@ def convert_coco(labels_dir='../coco/annotations/', use_segments=False, use_keyp
         Generates output files in the specified output directory.
     """
 
-    save_dir = make_dirs('yolo_labels')  # output directory
+    # Create dataset directory
+    save_dir = Path('yolo_labels')
+    if save_dir.exists():
+        shutil.rmtree(save_dir)  # delete dir
+    for p in save_dir / 'labels', save_dir / 'images':
+        p.mkdir(parents=True, exist_ok=True)  # make dir
+
+    # Convert classes
     coco80 = coco91_to_coco80_class()
 
     # Import json

ultralytics/data/loaders.py

@@ -16,7 +16,7 @@ import torch
 from PIL import Image
 
 from ultralytics.data.utils import IMG_FORMATS, VID_FORMATS
-from ultralytics.utils import LOGGER, ROOT, is_colab, is_kaggle, ops
+from ultralytics.utils import ASSETS, LOGGER, is_colab, is_kaggle, ops
 from ultralytics.utils.checks import check_requirements
 
 
@@ -403,7 +403,7 @@ def get_best_youtube_url(url, use_pafy=False):
 
 
 if __name__ == '__main__':
-    img = cv2.imread(str(ROOT / 'assets/bus.jpg'))
+    img = cv2.imread(str(ASSETS / 'bus.jpg'))
     dataset = LoadPilAndNumpy(im0=img)
     for d in dataset:
         print(d[0])