Image registration using non-linear diffusion