NekoCWD/libcamera
NekoCWD/libcamera
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 7 Wiki Activity Actions

libtuning: Fix visualize_macbeth_chart()

Browse source 
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>
This commit is contained in:
Stefan Klug 2024-06-11 08:59:17 +02:00
parent aa02706a34
commit 22c7c1e560
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
utils/tuning/libtuning/utils.py
View file

@ -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)