importtorchfromcolor_transferimportcolor_transfer_pytorch# In pytorch, you can run multiple images at once# but images must have the same sizesrc=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
Source
Target
Output
Benchmark results
The benchmark results can be reproduce by running python3 benchmark.py