from pylab import *
import datasets
ref, src = datasets.regframes('traffic')
import dtcwt
transform = dtcwt.Transform2d()
ref_t = transform.forward(ref, nlevels=6)
src_t = transform.forward(src, nlevels=6)
import dtcwt.registration as registration

reg = registration.estimatereg(src_t, ref_t)
warped_src = registration.warp(src, reg, method='bilinear')
figure()

imshow(np.dstack((ref, warped_src, np.zeros_like(ref))))
title('Source image warped to reference')