color-transfer-py 0.0.1

Color transfer between images

Color transfer between images

• Paper: Color Transfer between Images by Erik Reinhard, Michael Ashikhmin, Bruce Gooch and Peter Shirley.
• This is the re-implementation based on: https://github.com/chia56028/Color-Transfer-between-Images.
• The transfer process performed in numpy array instead of looping over pixel to improve performance.
• Support on pytorch and numpy.

Requirements

• numpy
• opencv-python
• torch (If you want to use pytorch version)

Usage

git clone https://github.com/ptran1203/color_transfer

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)

Transfer results

Source	Target	Output

Benchmark results

• The benchmark results can be reproduce by running python3 benchmark.py

This implementation	chia56028 implementation
0.02 second	1.337 second