image: adds image convertion tool

1 files changed, 29 insertions, 0 deletions

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)