HUB dataset previews to JPEG (#5627)

@kalenmike per our convo yesterday.
ultralytics · Nov 12, 2021 · d5b21b1 · d5b21b1
1 parent def7a0f
commit d5b21b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -965,7 +965,7 @@ def hub_ops(f, max_dim=1920):
             r = max_dim / max(im.height, im.width)  # ratio
             if r < 1.0:  # image too large
                 im = im.resize((int(im.width * r), int(im.height * r)))
-            im.save(f_new, quality=75)  # save
+            im.save(f_new, 'JPEG', quality=75, optimize=True)  # save
         except Exception as e:  # use OpenCV
             print(f'WARNING: HUB ops PIL failure {f}: {e}')
             im = cv2.imread(f)