def original_color_transform(content, generated, mask=None, hist_match=0, mode='YCbCr'): generated = fromimage(toimage(generated, mode='RGB'), mode=mode) # Convert to YCbCr color space if mask is None: if hist_match == 1: for channel in range(3): generated[:, :, channel] = match_histograms(generated[:, :, channel], content[:, :, channel]) else: generated[:, :, 1:] = content[:, :, 1:] else: width, height, channels = generated.shape for i in range(width): for j in range(height): if mask[i, j] == 1: if hist_match == 1: for channel in range(3): generated[i, j, channel] = match_histograms( generated[i, j, channel], content[i, j, channel]) else: generated[i, j, 1:] = content[i, j, 1:] generated = fromimage(toimage(generated, mode=mode), mode='RGB') # Convert to RGB color space return generated
def original_color_transform(content, generated, mask=None): generated = fromimage(toimage(generated, mode='RGB'), mode='YCbCr') # Convert to YCbCr color space if mask is None: generated[:, :, 1:] = content[:, :, 1:] # Generated CbCr = Content CbCr else: width, height, channels = generated.shape for i in range(width): for j in range(height): if mask[i, j] == 1: generated[i, j, 1:] = content[i, j, 1:] generated = fromimage(toimage(generated, mode='YCbCr'), mode='RGB') # Convert to RGB color space return generated