601d77a77a55323770bb2b214aa8844538b13d1f,art/visualization.py,,convert_to_rgb,#Any#,51

Before Change

    s = np.shape(images)

    rgb_images = np.zeros(shape=(s[0], s[1], s[2], 3))
    for i, img in enumerate(images):
        new_image = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)
        rgb_images[i] = new_image

    return rgb_images


def save_image(image, f_name):

After Change

    
    s = np.shape(images)
    if not ((len(s) == 4 and s[-1] == 1) or len(s) == 3):
        raise ValueError("Unexpected shape for grayscale images:" + str(s))

    if s[-1] == 1:
        // Squeeze channel axis if it exists

Instances