Pytorch affine transforms

reillytilbury / coppafish

Python version of iss software

MIT License

4 stars 2 forks source link

Pytorch affine transforms #254

Closed paulshuker closed 2 months ago

paulshuker commented 3 months ago

Instead of the slower scipy function affine_transform, use the faster (and GPU available) pytorch affine function
This will likely be of use: https://math.stackexchange.com/questions/237369/given-this-transformation-matrix-how-do-i-decompose-it-into-translation-rotati and https://www.eecs.qmul.ac.uk/~gslabaugh/publications/euler.pdf

paulshuker commented 2 months ago

This has been optimised by affine transforming positions using matrix multiplication then using the pytorch grid_sample function to gather optical flow shifts with interpolation.