7c57e6f2de0795edef5aed8cbcebb1243ba86191,lib/reconstruct.py,,image,#Any#Any#Any#Any#,126

Before Change

        dst = blockwise(src[:, :, 0], model, block_size, batch_size)
        dst = np.clip(dst, 0, 1) * 255
        src[:, :, 0] = dst[:, :, 0]
        dst = Image.fromarray(src, mode="YCbCr").convert("RGB")
    elif model.ch == 3:
        src = np.array(src.convert("RGB"), dtype=np.uint8)
        dst = blockwise(src, model, block_size, batch_size)

After Change

        dst = blockwise(src[:, :, 0], model, block_size, batch_size)
        dst = np.clip(dst, 0, 1) * 255
        src[:, :, 0] = dst[:, :, 0]
        dst = Image.fromarray(src, mode="YCbCr")
        if y2rgb:
            dst = dst.split()[0]
        else:
            dst = dst.convert("RGB")
    elif model.ch == 3:
        y2rgb = src.mode == "L"
        if y2rgb:
            src = iproc.y2rgb(src)

Instances