numpy array to image

numpy_array_to_image.py

from colorsys import hsv_to_rgb
from PIL import Image
import numpy as np

SAT_MIN = 10e-4
SAT_MAX = 10e-1

def clamp(val):
    return max(min(val, 1), 0)

def val_to_sat(val):
    out = min(max(val - SAT_MIN, 0)), SAT_MAX-SAT_MIN) / (SAT_MAX-SAT_MIN)
    return val

def np_arr_to_image(arr):
    colors = []
    height = arr.shape[0]
    width = arr.shape[1]

    for i in range(height):
        for j in range(width):
            val = arr[i][j]
            hue = 0
            sat = val_to_sat(val)
            rgb = hsv_to_rgb(hue, sat, sat)
            rgb = [int(255*clamp(u)) for u in rgb]
            colors.extend(rgb)
    colors = bytes(colors)
    return Image.frombytes('RGB', (height, width), colors)