Transforming a color image with astroalign

gistfile1.txt

from PIL import Image
import numpy as np
import astroalign as aa

source = np.array(Image.open("source.jpg"))
target = np.array(Image.open("target.jpg"))

def register_color(source, target):
    transf = aa.find_transform(source[:,:,1], target[:,:,1])[0]
    rgb = []
    for achannel in np.rollaxis(source, 2):
        rgb.append(aa.apply_transform(transf, achannel, target[:,:,1]))
    return np.array(rgb)

transformed = register_color(source, target)