Spaces:
Sleeping
Sleeping
| #!/usr/bin/env python3 | |
| # this routine reads in many image files which are 128 pixels high | |
| # and finds the lowest and highest non-black pixels across all images | |
| import sys | |
| from PIL import Image | |
| from control_toys.data import fast_scandir | |
| from tqdm import tqdm | |
| if __name__ == "__main__": | |
| if len(sys.argv) < 2: | |
| print(f"Usage: {sys.argv[0]} <image-dirs>") | |
| sys.exit(1) | |
| img_dirs = sys.argv[1:] | |
| img_files = [] | |
| for imgdir in img_dirs: | |
| i_subdirs, imfile = fast_scandir(imgdir, ['png', 'jpg', 'jpeg']) | |
| if imfile != []: img_files = img_files + imfile | |
| # remove any files with the 'transpose' in the name | |
| img_files = [f for f in img_files if 'transpose' not in f] | |
| if len(img_files) == 0: | |
| print(f"No image files found in {img_dirs}") | |
| sys.exit(1) | |
| min_note = 128 | |
| max_note = 0 | |
| for img_file in tqdm(img_files): | |
| img = Image.open(img_file) | |
| img = img.convert('L') | |
| for y in range(img.size[-1]): | |
| for x in range(img.size[0]): | |
| val = img.getpixel((x,y)) | |
| if val > 0: | |
| if y < min_note: min_note = y | |
| if y > max_note: max_note = y | |
| break | |
| print(f"min_note = {min_note}, max_note = {max_note}") | |
| transpose_max = 12 | |
| print(f"So with transpostion of +/-{transpose_max}, the range is", min_note-transpose_max, "to", max_note+transpose_max) | |
| sys.exit(0) |