use pillow to save higher-quality jpg (w/o color subsampling)

ultralytics · Mar 16, 2022 · f334965 · f334965 · LaserBorg · Mar 16, 2022
1 parent c09fb2a
commit f334965
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/utils/plots.py b/utils/plots.py
@@ -467,5 +467,7 @@ def save_one_box(xyxy, im, file='image.jpg', gain=1.02, pad=10, square=False, BG
     crop = im[int(xyxy[0, 1]):int(xyxy[0, 3]), int(xyxy[0, 0]):int(xyxy[0, 2]), ::(1 if BGR else -1)]
     if save:
         file.parent.mkdir(parents=True, exist_ok=True)  # make directory
-        cv2.imwrite(str(increment_path(file).with_suffix('.jpg')), crop)
+        # use pillow to save higher-quality jpg (w/o color subsampling)
+        crop_pil = Image.fromarray(cv2.cvtColor(crop, cv2.COLOR_BGR2RGB))
+        crop_pil.save(str(increment_path(file).with_suffix('.jpg')), quality=95, subsampling=0)
     return crop