diff --git a/docs/hub/datasets.md b/docs/hub/datasets.md index e2ac4654..d5e23443 100644 --- a/docs/hub/datasets.md +++ b/docs/hub/datasets.md @@ -23,7 +23,7 @@ For example, if your dataset is called "coco8", as our [COCO8](https://docs.ultr zip -r coco8.zip coco8 ``` -You can download our [COCO8](https://github.com/ultralytics/hub/blob/master/example_datasets/coco8.zip) example dataset and unzip it to see exactly how to structure your dataset. +You can download our [COCO8](https://github.com/ultralytics/hub/blob/main/example_datasets/coco8.zip) example dataset and unzip it to see exactly how to structure your dataset.
diff --git a/tests/test_python.py b/tests/test_python.py
index 2a0d556f..1fc5fc81 100644
--- a/tests/test_python.py
+++ b/tests/test_python.py
@@ -275,7 +275,7 @@ def test_data_utils():
# from ultralytics.utils.files import WorkingDirectory
# with WorkingDirectory(ROOT.parent / 'tests'):
- download('https://github.com/ultralytics/hub/raw/master/example_datasets/coco8.zip', unzip=False)
+ download('https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip', unzip=False)
shutil.move('coco8.zip', TMP)
stats = HUBDatasetStats(TMP / 'coco8.zip', task='detect')
stats.get_json(save=True)
diff --git a/ultralytics/data/utils.py b/ultralytics/data/utils.py
index 6bc2e30d..05b291ba 100644
--- a/ultralytics/data/utils.py
+++ b/ultralytics/data/utils.py
@@ -373,6 +373,7 @@ class HUBDatasetStats:
autodownload (bool): Attempt to download dataset if not found locally. Default is False.
Example:
+ Download *.zip files from i.e. https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip.
```python
from ultralytics.data.utils import HUBDatasetStats