xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update utils.py

This commit is contained in:
Hardik Dava 2023-08-18 12:42:23 +02:00 committed by GitHub
parent 98f12b60ea
commit b40eb5b47f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
supervision/dataset/utils.py
View File

@ -97,8 +97,10 @@ def save_dataset_images(
) -> None:
Path(images_directory_path).mkdir(parents=True, exist_ok=True)
for image_name, image in images.items():
target_image_path = os.path.join(images_directory_path, image_name)
for image_path, image in images.items():
image_name = str(Path(image_path).stem)
image_ext = str(Path(image_path).suffix)
target_image_path = os.path.join(images_directory_path, f"{image_name}{image_ext}")
cv2.imwrite(target_image_path, image)