rewrite

data/main.py

@@ -0,0 +1,39 @@
+import argparse
+
+import cv2
+import numpy as np
+
+def parse_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--input", required=True, help="input image path")
+    parser.add_argument("--output", required=False, default='output.h', help="path to output file")
+
+    return parser.parse_args()
+
+
+def main():
+    args = parse_args()
+    image = cv2.imread(args.input, cv2.IMREAD_GRAYSCALE)
+    if image is None:
+        print(
+            f"Could not read image {args.input}. "
+            f"Please, check that you passed the correct path"
+        )
+        exit()
+
+    image = cv2.resize(image, (128, 48))
+    image[image > 0] = 1
+    image = 1 - image
+    image = list(np.ravel(image))
+    image_text = ''
+    for i, pixel_data in enumerate(image):
+        image_text += str(pixel_data)
+        if i != len(image) - 1:
+            image_text += ', '
+    text = f"#ifndef _OUTPUT_\n#define _OUTPUT_\nbyte output[128*64] = {{ {image_text} }};\n\n#endif"
+
+    with open(args.output, 'w') as f:
+        f.write(text)
+
+if __name__ == "__main__":
+    main()