From d5b21b1ecb66b35af937ac12364aa80733222bd1 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 12 Nov 2021 12:05:25 +0100
Subject: [PATCH] HUB dataset previews to JPEG (#5627)

@kalenmike per our convo yesterday.
---
 utils/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index 1ecc744..2a6653b 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -965,7 +965,7 @@ def dataset_stats(path='coco128.yaml', autodownload=False, verbose=False, profil
             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)