7d8f3f66495979f1bbd27205d422d673991709f2,GANs/stylegan2/ops.py,,convert_images_to_uint8,#,123

Before Change

    if isinstance(images, nn.Variable):
        images = images.d

    if images.ndim == 4:
        images = images[0]

    scale = 255 / (drange[1] - drange[0])
    images = images * scale + (0.5 - drange[0] * scale)
    return np.uint8(np.clip(images, 0, 255))

After Change

    
    if isinstance(images, nn.Variable):
        images = images.d
    if isinstance(images, nn.NdArray):
        images = images.data

    scale = 255 / (drange[1] - drange[0])
    images = images * scale + (0.5 - drange[0] * scale)
    return np.uint8(np.clip(images, 0, 255))

Instances