xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Avoid creating -hub dir without processing images (#8334)

Browse Source
This commit is contained in:
Glenn Jocher 2024-02-20 17:27:28 +01:00 committed by GitHub
parent 61088b43d7
commit d03928f371
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ultralytics/data/utils.py
View File

@ -467,7 +467,6 @@ class HUBDatasetStats:
self.hub_dir = Path(f'{data["path"]}-hub') self.hub_dir = Path(f'{data["path"]}-hub')
self.im_dir = self.hub_dir / "images" self.im_dir = self.hub_dir / "images"
self.im_dir.mkdir(parents=True, exist_ok=True) # makes /images
self.stats = {"nc": len(data["names"]), "names": list(data["names"].values())} # statistics dictionary self.stats = {"nc": len(data["names"]), "names": list(data["names"].values())} # statistics dictionary
self.data = data self.data = data
@ -551,6 +550,7 @@ class HUBDatasetStats:
# Save, print and return # Save, print and return
if save: if save:
self.hub_dir.mkdir(parents=True, exist_ok=True) # makes dataset-hub/
stats_path = self.hub_dir / "stats.json" stats_path = self.hub_dir / "stats.json"
LOGGER.info(f"Saving {stats_path.resolve()}...") LOGGER.info(f"Saving {stats_path.resolve()}...")
with open(stats_path, "w") as f: with open(stats_path, "w") as f:
@ -563,6 +563,7 @@ class HUBDatasetStats:
"""Compress images for Ultralytics HUB.""" """Compress images for Ultralytics HUB."""
from ultralytics.data import YOLODataset # ClassificationDataset from ultralytics.data import YOLODataset # ClassificationDataset
self.im_dir.mkdir(parents=True, exist_ok=True) # makes dataset-hub/images/
for split in "train", "val", "test": for split in "train", "val", "test":
if self.data.get(split) is None: if self.data.get(split) is None:
continue continue