Thin Plate Spline implementation with PyTorch
Project description
torch-tps
Implementation of Thin Plate Spline. (See numpy implementation with thin-plate-spline library)
Install
Pip
$ pip install torch-tps
Conda
Not yet available
Getting started
import torch
from tps import ThinPlateSpline
# Some data
X_c = torch.normal(0, 1, (800, 3))
X_t = torch.normal(0, 2, (800, 2))
X = torch.normal(0, 1, (300, 3))
# Create the tps object
tps = ThinPlateSpline(alpha=0.0) # 0 Regularization
# Fit the control and target points
tps.fit(X_c, X_t)
# Transform new points
Y = tps.transform(X)
Also have a look at example.py
Build and Deploy
$ pip install build twine
$ python -m build
$ python -m twine upload dist/*
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torch-tps-1.0.0.tar.gz
(4.3 kB
view hashes)
Built Distribution
Close
Hashes for torch_tps-1.0.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d2ec99a25409b8eac73c55c9291b1065cb53f05265ce8f025b2f890f49a54674 |
|
MD5 | 5d100de7387ab343f8cc3ae15192f01b |
|
BLAKE2b-256 | 589568dfa4173eb05eaccd0c2ccf731e3dfba2809c4192f5cc6d9b1663229d25 |