Home

Awesome

Color transfer between images

Requirements

Usage

pip install color_transfer_py

Numpy usage

import cv2
from color_transfer import color_transfer

src = cv2.imread("examples/1/src.jpeg")
target = cv2.imread("examples/1/target.png")

out = color_transfer(src, target)

cv2.imwrite("examples/1/output.jpeg", out)

Pytorch usage

import torch
from color_transfer import color_transfer_pytorch

# In pytorch, you can run multiple images at once
# but images must have the same size

src = torch.rand(4, 3, 224, 224)
target = torch.rand(4, 3, 224, 224)

outs = color_transfer_pytorch(src, target)
# Outs shape: (4, 3, 224, 224), output range is [0, 1]

Transfer results

SourceTargetOutput
c1g1g1
c1g1g1
c1g1g1
c1g1g1

Benchmark results

This implementationchia56028 implementation
0.02 second1.337 second