fix(pre_commit): 🎨 auto format pre-commit hooks

This commit is contained in:
pre-commit-ci[bot] 2023-08-18 10:39:48 +00:00
parent 2c988ac269
commit 98f12b60ea
1 changed files with 3 additions and 1 deletions

View File

@ -626,7 +626,9 @@ class ClassificationDataset(BaseDataset):
else classification.get_top_k(1)[0][0]
)
class_name = self.classes[class_id]
image_path = os.path.join(root_directory_path, class_name, f"{image_name}.{image_ext}")
image_path = os.path.join(
root_directory_path, class_name, f"{image_name}.{image_ext}"
)
cv2.imwrite(image_path, image)
@classmethod