def save_image_from_tensor(image_array, path):
    og = Image.fromarray(image_array.numpy())
    og = og.convert('RGB')
    og.save(path)
def save_image_from_tensor(image_array, path):
    og = Image.fromarray(image_array.numpy())
    og = og.convert('RGB')
    og.save(path)