From d03928f37174ff9f2285824d3b56b3e105f449b7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 20 Feb 2024 17:27:28 +0100 Subject: [PATCH] Avoid creating `-hub` dir without processing images (#8334) --- ultralytics/data/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ultralytics/data/utils.py b/ultralytics/data/utils.py index 9214f8ce..be331bf3 100644 --- a/ultralytics/data/utils.py +++ b/ultralytics/data/utils.py @@ -467,7 +467,6 @@ class HUBDatasetStats: self.hub_dir = Path(f'{data["path"]}-hub') 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.data = data @@ -551,6 +550,7 @@ class HUBDatasetStats: # Save, print and return if save: + self.hub_dir.mkdir(parents=True, exist_ok=True) # makes dataset-hub/ stats_path = self.hub_dir / "stats.json" LOGGER.info(f"Saving {stats_path.resolve()}...") with open(stats_path, "w") as f: @@ -563,6 +563,7 @@ class HUBDatasetStats: """Compress images for Ultralytics HUB.""" 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": if self.data.get(split) is None: continue