Skip to content

Cuda implementation for gridsample with second derivative support

License

Notifications You must be signed in to change notification settings

AliaksandrSiarohin/cuda-gridsample-grad2

Repository files navigation

Cuda-based grid sample implimentation with second order derivative support.

For the forward and backward pass it uses the pytorch version of grid sample. Only the second backward is implimented here.

Tested only with Pytorch 10.1 and 10.2

Reflection padding is not supported

Usage:

import cuda_gridsample as cu

cu.grid_sample_2d(image, optical, padding_mode='border', align_corners=True)
cu.grid_sample_3d(volume, optical, padding_mode='border', align_corners=True)

About

Cuda implementation for gridsample with second derivative support

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published