You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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