update comment

sdbds · Nov 11, 2024 · 3fe94b0 · 3fe94b0
1 parent 92482c7
commit 3fe94b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/library/train_util.py b/library/train_util.py
@@ -1404,7 +1404,7 @@ def get_image_size(self, image_path):
         # return imagesize.get(image_path)
         image_size = imagesize.get(image_path)
         if image_size[0] <= 0:
-            # imagesize doesn't work for some images, so use cv2
+            # imagesize doesn't work for some images, so use PIL as a fallback
             try:
                 with Image.open(image_path) as img:
                     image_size = img.size