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

    if s[-1] == 1:
        // Squeeze channel axis if it exists
        rgb_images = np.squeeze(images, axis=-1)
    else:
        rgb_images = images
    rgb_images = np.stack((rgb_images,) * 3, axis=-1)

    return rgb_images

Instances