diff --git a/docs/yolov5/tutorials/train_custom_data.md b/docs/yolov5/tutorials/train_custom_data.md index 25de78c..ee8ca2d 100644 --- a/docs/yolov5/tutorials/train_custom_data.md +++ b/docs/yolov5/tutorials/train_custom_data.md @@ -30,12 +30,12 @@ Creating a custom model to detect your objects is an iterative process of collec YOLOv5 models must be trained on labelled data in order to learn classes of objects in that data. There are two options for creating your dataset before you start training: -
-Use Roboflow to create your dataset in YOLO format +
+Use Roboflow to create your dataset in YOLO format 🌟 -!!! warning +!!! note - Roboflow users can use Ultralytics under the [AGPL license](https://github.com/ultralytics/ultralytics/blob/main/LICENSE) or procure an [Enterprise license](https://ultralytics.com/license) directly from Ultralytics. Be aware that Roboflow does **not** provide Ultralytics licenses, and it is the responsibility of the user to ensure appropriate licensing. + Roboflow users can use Ultralytics under the [AGPL license](https://github.com/ultralytics/ultralytics/blob/main/LICENSE) or can request an [Enterprise license](https://ultralytics.com/license) directly from Ultralytics. Be aware that Roboflow does not provide Ultralytics licenses, and it is the responsibility of the user to ensure appropriate licensing. ### 1.1 Collect Images @@ -84,7 +84,7 @@ script or notebook to download your dataset. Now continue with `2. Select a Model`.
-
+
Or manually prepare your dataset ### 1.1 Create dataset.yaml diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index a507371..ee89764 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.137' +__version__ = '8.0.138' from ultralytics.engine.model import YOLO from ultralytics.hub import start diff --git a/ultralytics/hub/utils.py b/ultralytics/hub/utils.py index adf0b3f..5487872 100644 --- a/ultralytics/hub/utils.py +++ b/ultralytics/hub/utils.py @@ -13,6 +13,7 @@ from tqdm import tqdm from ultralytics.utils import (ENVIRONMENT, LOGGER, ONLINE, RANK, SETTINGS, TESTS_RUNNING, TQDM_BAR_FORMAT, TryExcept, __version__, colorstr, get_git_origin_url, is_colab, is_git_dir, is_pip_package) +from ultralytics.utils.downloads import GITHUB_ASSET_NAMES PREFIX = colorstr('Ultralytics HUB: ') HELP_MSG = 'If this issue persists please visit https://github.com/ultralytics/hub/issues for assistance.' @@ -193,7 +194,9 @@ class Events: # Attempt to add to events if len(self.events) < 25: # Events list limited to 25 events (drop any events past this) - params = {**self.metadata, **{'task': cfg.task}} + params = { + **self.metadata, 'task': cfg.task, + 'model': cfg.model if cfg.model in GITHUB_ASSET_NAMES else 'custom'} if cfg.mode == 'export': params['format'] = cfg.format self.events.append({'name': cfg.mode, 'params': params})