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
The Tidelift Subscription provides access to a continuously curated stream of human-researched and maintainer-verified data on open source packages and their licenses, releases, vulnerabilities, and development practices.