/coupe.pytorch_tps

Thin plate spline deformation module for PyTorch

Primary LanguagePythonMIT LicenseMIT

Thin Plate Spline interpolation for PyTorch

Warps 2d PyTorch image (NxCxHxW) according to sparse 2d control points

image image

  • Mesh surface position sampling from barycentric coordinates for PyTorch

Build

Just copy and paste the code, or run python setup.py build && python setup.py install