From 22c7c1e560ded2f0bf882cc92283881bdeedc2cc Mon Sep 17 00:00:00 2001
From: Stefan Klug <stefan.klug@ideasonboard.com>
Date: Tue, 11 Jun 2024 08:59:17 +0200
Subject: libtuning: Fix visualize_macbeth_chart()
The old function uses PIL to save the image, which is not in the
requirements file. As we are already requiring opencv, use that to save
images instead of an additional dependency
Signed-off-by: Stefan Klug <stefan.klug@ideasonboard.com>
Reviewed-by: Paul Elder <paul.elder@ideasonboard.com>
Reviewed-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
---
utils/tuning/libtuning/utils.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
(limited to 'utils/tuning/libtuning')
diff --git a/utils/tuning/libtuning/utils.py b/utils/tuning/libtuning/utils.py
index 87234140..90fd7072 100644
--- a/utils/tuning/libtuning/utils.py
+++ b/utils/tuning/libtuning/utils.py
@@ -5,6 +5,7 @@
#
# Utilities for libtuning
+import cv2
import decimal
import math
import numpy as np
@@ -162,6 +163,6 @@ def visualise_macbeth_chart(macbeth_rgb, original_rgb, new_rgb, output_filename)
for g in range(100):
image[xlocation + i, ylocation + g] = new_rgb[colorindex]
- img = Image.fromarray(image, 'RGB')
- img.save(str(output_filename) + 'Generated Macbeth Chart.png')
+ im_bgr = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+ cv2.imwrite(f'{output_filename} Generated Macbeth Chart.png', im_bgr)
--
cgit v1.2.1