From 1d42992e6b2edaf4c86df10064ea237d9fb0fba7 Mon Sep 17 00:00:00 2001 From: Hardik Dava <39372750+hardikdava@users.noreply.github.com> Date: Fri, 18 Aug 2023 12:44:35 +0200 Subject: [PATCH] Update core.py --- supervision/dataset/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/dataset/core.py b/supervision/dataset/core.py index 9143164c..fcbaad0c 100644 --- a/supervision/dataset/core.py +++ b/supervision/dataset/core.py @@ -627,7 +627,7 @@ class ClassificationDataset(BaseDataset): ) class_name = self.classes[class_id] image_path = os.path.join( - root_directory_path, class_name, f"{image_name}.{image_ext}" + root_directory_path, class_name, f"{image_name}{image_ext}" ) cv2.imwrite(image_path, image)