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
$ 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.2.tar.gz
(4.4 kB
view hashes)
Built Distribution
Close
Hashes for torch_tps-1.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 443f9a4d4757f527a629ceed1e32834331e7e54647842913bbcca98e90695274 |
|
MD5 | 7dd8ce00247adee88bd377500b10607c |
|
BLAKE2b-256 | 5fa5e47a74f348919ce858a23f1c95c756a99e9a2c6797ef6d183ece5780ff1b |