nenuscanner/calibration.py

30 lines
577 B
Python
Executable File

#!/usr/bin/env python
import functools
import numpy as np
import os
import sys
from PIL import Image
def print_error(msg: str) -> None:
print('\x1b[1;31m[ERR]' + msg + '\x1b[0m', file=sys.stderr)
def main():
if len(sys.argv) < 2:
print_error('Expected path to images as argument')
sys.exit(1)
# Load images
input_dir = sys.argv[1]
images = [np.asarray(Image.open(os.path.join(input_dir, x))) for x in os.listdir(input_dir)]
# Max image
max_image = functools.reduce(np.maximum, images)
if __name__ == '__main__':
main()