Skip to content

Subpixel phase correlation for image registration, adapted for Pytorch API with GPU support

Notifications You must be signed in to change notification settings

mdw771/pytorch_phase_cross_correlation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Subpixel phase correlation for translational image registration, adapted from skimage.registration.phase_cross_correlation for Pytorch API with GPU support.

The function allows for complex input. In that case, the real part and imaginary part of each image should be held in the last dimension of the image tensor - e.g., the shape of the image tensor should be [y, x, 2].

Note: This function may not work with automatic differentiation. I use it only for fast image registration on GPU outside of the graph.

About

Subpixel phase correlation for image registration, adapted for Pytorch API with GPU support

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages