Faster crop saving (#8696)

Faster crops

Following https://github.com/ultralytics/yolov5/issues/8641#issuecomment-1193190325
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-07-23 23:30:30 +02:00
1 parent b510957650
commit 916bdb1d61
1 file changed
+2 -2
+2 -2
View File
@@ -484,6 +484,6 @@ def save_one_box(xyxy, im, file=Path('im.jpg'), gain=1.02, pad=10, square=False,
if save:
file.parent.mkdir(parents=True, exist_ok=True) # make directory
f = str(increment_path(file).with_suffix('.jpg'))
# cv2.imwrite(f, crop) # https://github.com/ultralytics/yolov5/issues/7007 chroma subsampling issue
Image.fromarray(cv2.cvtColor(crop, cv2.COLOR_BGR2RGB)).save(f, quality=95, subsampling=0)
# cv2.imwrite(f, crop) # save BGR, https://github.com/ultralytics/yolov5/issues/7007 chroma subsampling issue
Image.fromarray(crop[..., ::-1]).save(f, quality=95, subsampling=0) # save RGB
return crop