def transform_img_for_model(image_array, transforms=None):
    image_array_copy = np.copy(image_array)
    image_array_copy = np.expand_dims(image_array_copy, axis=2)
    if (transforms is None):
        image_array_copy = torch.from_numpy(image_array_copy).repeat(3, 1, 1)
    else:
        image_array_copy = transforms(image_array_copy).repeat(3, 1, 1)
    return image_array_copy
def transform_img_for_model(image_array, transforms=None):
    image_array_copy = np.copy(image_array)
    image_array_copy = np.expand_dims(image_array_copy, axis=2)
    if (transforms is None):
        image_array_copy = torch.from_numpy(image_array_copy).repeat(3, 1, 1)
    else:
        image_array_copy = transforms(image_array_copy).repeat(3, 1, 1)
    return image_array_copy