Deep learning on point clouds, with PyTorch.
Models, pretrained weights, datasets, transforms and inferers, in the style of
timm and
PyG.
Documentation · Model zoo · Tutorials · Changelog
Installation
pip install torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu
pip install torch-pointcloud
pip install pyg-lib -f https://data.pyg.org/whl/torch-2.10.0+cpu.html
Most models also need pyg-lib, and some a CUDA extension (spconv, flash-attn, ...), built for your torch and
CUDA versions. Read the installation documentation for further details.
Quickstart
import torch
import torch_pointcloud as tp
# Requires pyg-lib
model = tp.create_model(
"pointnext-sm.scanobjectnn-hardest.openpoints",
task="classification",
pretrained=True,
).eval()
pos = torch.randn(2048, 3) # (N, 3) coordinates
x = torch.cat([pos, pos[:, 1:2] - pos[:, 1].min()], dim=1) # (N, 4) features: xyz + height
batch = torch.zeros(2048, dtype=torch.long) # (N,) batch index
with torch.no_grad():
logits = model(x, pos, batch) # (1, 15)
Each checkpoint ships information about the model, such as the transform used to preprocess the input, metrics, license and more.
import torch_pointcloud as tp
model, info = tp.create_model("randlanet.semantickitti.tsung-han-wu", task="semantic-segmentation", pretrained=True, return_info=True)
info["transform"] # the preprocessing pipeline of that checkpoint
info["weights"]["metrics"] # {"mIoU": 55.39, "OA": 90.05}
tp.list_models("pointnext*") # every registered PointNeXt config
tp.list_models(task="detection", pretrained=True) # all detection checkpoints
See the examples directory for benchmarks and training recipes.
Why torch-pointcloud?
torch-pointcloud is a library of pretrained models that makes common layers and backbones easy to reuse. It does
not replace research-first codebases such as
Pointcept, OpenPCDet and
MMDetection3D or OpenPoints
but it complements them with a common interface that makes benchmarking and interoperability across architectures easier.
Contributing
Contributions are welcome. See CONTRIBUTING.md for the development setup and the pull request checks.
Citation
If you find this project useful, please consider citing:
@software{dujardin2026pytorchpointcloud,
author = {Dujardin, Arthur},
title = {PyTorch PointCloud},
year = {2026},
url = {https://github.com/arthurdjn/pytorch-pointcloud},
doi = {10.5281/zenodo.22159632},
license = {Apache-2.0}
}
License
Apache 2.0. See LICENSE.
Pretrained weights and adapted code keep the license of their source, and some checkpoints are restricted to
non-commercial use (create_model warns when loading them). Most were trained on research-only datasets, whose terms
also apply to the weights.
See THIRD_PARTY_NOTICES.md.
1 maintainer
Metadata
1 maintainer
Release files for torch-pointcloud 0.0.18
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | Provenance | |
|---|---|---|---|---|
| torch_pointcloud-0.0.18.tar.gz | 601.1 kB | Without provenance | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| torch_pointcloud-0.0.18-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 1.3 MB
Release files / torch_pointcloud-0.0.18.tar.gz
| Download URL | torch_pointcloud-0.0.18.tar.gz |
|---|---|
| Size | 601.1 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
292dd1e0fda82f695754052a442fa3ffb15788a6ff7f500052cc3abe50b2a194
|
|
BLAKE2b-256 checksum How to use checksums |
8c0f2718d8e65e5b40e14aa7711a2be8b89be90f8d7fcb0da9696e06b2baf302
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.13.4 {"installer":{"name":"uv","version":"0.13.4","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
Release files / torch_pointcloud-0.0.18-py3-none-any.whl
| Download URL | torch_pointcloud-0.0.18-py3-none-any.whl |
|---|---|
| Size | 735.1 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
a90188afd99ca8d311eb3972ea412173c41f1aecffcfdc0aa7c53b051f614ef5
|
|
BLAKE2b-256 checksum How to use checksums |
a59908ee139512dcd3a8e50d0b8be3289f1b6ece62f11f1a88d25517f3b612ce
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.13.4 {"installer":{"name":"uv","version":"0.13.4","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
Security for torch-pointcloud 0.0.18
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
No provenance found
None of the files in this release were uploaded with provenance metadata.
A note on package security
PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.
To learn more about protecting yourself from malicious actors, visit our package security documentation.
