New ASSETS and trackers GMC cleanup (#4425)

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

ultralytics/models/yolo/classify/predict.py

@@ -4,7 +4,7 @@ import torch
 
 from ultralytics.engine.predictor import BasePredictor
 from ultralytics.engine.results import Results
-from ultralytics.utils import DEFAULT_CFG, ROOT
+from ultralytics.utils import ASSETS, DEFAULT_CFG
 
 
 class ClassificationPredictor(BasePredictor):
@@ -35,8 +35,7 @@ class ClassificationPredictor(BasePredictor):
 def predict(cfg=DEFAULT_CFG, use_python=False):
     """Run YOLO model predictions on input images/videos."""
     model = cfg.model or 'yolov8n-cls.pt'  # or "resnet18"
-    source = cfg.source if cfg.source is not None else ROOT / 'assets' if (ROOT / 'assets').exists() \
-        else 'https://ultralytics.com/images/bus.jpg'
+    source = cfg.source or ASSETS
 
     args = dict(model=model, source=source)
     if use_python:

ultralytics/models/yolo/detect/predict.py

@@ -4,7 +4,7 @@ import torch
 
 from ultralytics.engine.predictor import BasePredictor
 from ultralytics.engine.results import Results
-from ultralytics.utils import DEFAULT_CFG, ROOT, ops
+from ultralytics.utils import ASSETS, DEFAULT_CFG, ops
 
 
 class DetectionPredictor(BasePredictor):
@@ -32,8 +32,7 @@ class DetectionPredictor(BasePredictor):
 def predict(cfg=DEFAULT_CFG, use_python=False):
     """Runs YOLO model inference on input image(s)."""
     model = cfg.model or 'yolov8n.pt'
-    source = cfg.source if cfg.source is not None else ROOT / 'assets' if (ROOT / 'assets').exists() \
-        else 'https://ultralytics.com/images/bus.jpg'
+    source = cfg.source or ASSETS
 
     args = dict(model=model, source=source)
     if use_python:

ultralytics/models/yolo/detect/train.py

@@ -107,7 +107,7 @@ class DetectionTrainer(BaseTrainer):
 def train(cfg=DEFAULT_CFG, use_python=False):
     """Train and optimize YOLO model given training data and device."""
     model = cfg.model or 'yolov8n.pt'
-    data = cfg.data or 'coco128.yaml'  # or yolo.ClassificationDataset("mnist")
+    data = cfg.data or 'coco8.yaml'  # or yolo.ClassificationDataset("mnist")
     device = cfg.device if cfg.device is not None else ''
 
     args = dict(model=model, data=data, device=device)

ultralytics/models/yolo/detect/val.py

@@ -6,7 +6,7 @@ from pathlib import Path
 import numpy as np
 import torch
 
-from ultralytics.data import build_dataloader, build_yolo_dataset
+from ultralytics.data import build_dataloader, build_yolo_dataset, converter
 from ultralytics.engine.validator import BaseValidator
 from ultralytics.utils import DEFAULT_CFG, LOGGER, ops
 from ultralytics.utils.checks import check_requirements
@@ -50,7 +50,7 @@ class DetectionValidator(BaseValidator):
         """Initialize evaluation metrics for YOLO."""
         val = self.data.get(self.args.split, '')  # validation path
         self.is_coco = isinstance(val, str) and 'coco' in val and val.endswith(f'{os.sep}val2017.txt')  # is COCO
-        self.class_map = ops.coco80_to_coco91_class() if self.is_coco else list(range(1000))
+        self.class_map = converter.coco80_to_coco91_class() if self.is_coco else list(range(1000))
         self.args.save_json |= self.is_coco and not self.training  # run on final val if training COCO
         self.names = model.names
         self.nc = len(model.names)
@@ -259,7 +259,7 @@ class DetectionValidator(BaseValidator):
 def val(cfg=DEFAULT_CFG, use_python=False):
     """Validate trained YOLO model on validation dataset."""
     model = cfg.model or 'yolov8n.pt'
-    data = cfg.data or 'coco128.yaml'
+    data = cfg.data or 'coco8.yaml'
 
     args = dict(model=model, data=data)
     if use_python:

ultralytics/models/yolo/pose/predict.py

@@ -2,7 +2,7 @@
 
 from ultralytics.engine.results import Results
 from ultralytics.models.yolo.detect.predict import DetectionPredictor
-from ultralytics.utils import DEFAULT_CFG, LOGGER, ROOT, ops
+from ultralytics.utils import ASSETS, DEFAULT_CFG, LOGGER, ops
 
 
 class PosePredictor(DetectionPredictor):
@@ -45,8 +45,7 @@ class PosePredictor(DetectionPredictor):
 def predict(cfg=DEFAULT_CFG, use_python=False):
     """Runs YOLO to predict objects in an image or video."""
     model = cfg.model or 'yolov8n-pose.pt'
-    source = cfg.source if cfg.source is not None else ROOT / 'assets' if (ROOT / 'assets').exists() \
-        else 'https://ultralytics.com/images/bus.jpg'
+    source = cfg.source or ASSETS
 
     args = dict(model=model, source=source)
     if use_python:

ultralytics/models/yolo/segment/predict.py

@@ -4,7 +4,7 @@ import torch
 
 from ultralytics.engine.results import Results
 from ultralytics.models.yolo.detect.predict import DetectionPredictor
-from ultralytics.utils import DEFAULT_CFG, ROOT, ops
+from ultralytics.utils import ASSETS, DEFAULT_CFG, ops
 
 
 class SegmentationPredictor(DetectionPredictor):
@@ -47,8 +47,7 @@ class SegmentationPredictor(DetectionPredictor):
 def predict(cfg=DEFAULT_CFG, use_python=False):
     """Runs YOLO object detection on an image or video source."""
     model = cfg.model or 'yolov8n-seg.pt'
-    source = cfg.source if cfg.source is not None else ROOT / 'assets' if (ROOT / 'assets').exists() \
-        else 'https://ultralytics.com/images/bus.jpg'
+    source = cfg.source or ASSETS
 
     args = dict(model=model, source=source)
     if use_python:

ultralytics/models/yolo/segment/train.py

@@ -49,7 +49,7 @@ class SegmentationTrainer(yolo.detect.DetectionTrainer):
 def train(cfg=DEFAULT_CFG, use_python=False):
     """Train a YOLO segmentation model based on passed arguments."""
     model = cfg.model or 'yolov8n-seg.pt'
-    data = cfg.data or 'coco128-seg.yaml'  # or yolo.ClassificationDataset("mnist")
+    data = cfg.data or 'coco8-seg.yaml'
     device = cfg.device if cfg.device is not None else ''
 
     args = dict(model=model, data=data, device=device)

ultralytics/models/yolo/segment/val.py

@@ -236,7 +236,7 @@ class SegmentationValidator(DetectionValidator):
 def val(cfg=DEFAULT_CFG, use_python=False):
     """Validate trained YOLO model on validation data."""
     model = cfg.model or 'yolov8n-seg.pt'
-    data = cfg.data or 'coco128-seg.yaml'
+    data = cfg.data or 'coco8-seg.yaml'
 
     args = dict(model=model, data=data)
     if use_python: