From 14a3611e492d2f213e81c9053bf613a5d8ad30a6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabi=C3=A1n=20Montero?= Date: Wed, 16 Nov 2022 16:44:42 -0600 Subject: image: adds image convertion tool --- image_processing/array2rgb32.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100755 image_processing/array2rgb32.py (limited to 'image_processing/array2rgb32.py') diff --git a/image_processing/array2rgb32.py b/image_processing/array2rgb32.py new file mode 100755 index 0000000..167aa74 --- /dev/null +++ b/image_processing/array2rgb32.py @@ -0,0 +1,29 @@ +#!/usr/bin/env python3 + +import sys +import numpy as np +from PIL import Image +import matplotlib.pyplot as plt + +def get_base_image(image): + image = Image.fromarray(np.uint8(image)) + image = image.convert(mode="RGBA", colors=256) + + if image.size != (640, 480): # width, height + image = image.resize(size=(640, 480)) + + return image + +def show_image(image): + plt.imshow(image) + plt.show() + +image = get_base_image(eval(input(""))) +image_bytes = image.tobytes() + +#show_image(image) + +out_file, = sys.argv[1:] + +with open(out_file, 'wb') as f: + f.write(image_bytes) -- cgit v1.2.3