https://github.com/tchaton/torch-points3d-templates
Implement simple template for torch-points3d
https://github.com/tchaton/torch-points3d-templates
Last synced: 6 months ago
JSON representation
Implement simple template for torch-points3d
- Host: GitHub
- URL: https://github.com/tchaton/torch-points3d-templates
- Owner: tchaton
- Created: 2020-05-08T14:07:28.000Z (about 6 years ago)
- Default Branch: master
- Last Pushed: 2020-05-09T13:47:26.000Z (about 6 years ago)
- Last Synced: 2025-01-21T03:24:51.504Z (over 1 year ago)
- Language: Dockerfile
- Size: 78.1 KB
- Stars: 4
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Templates collection for torch-points3
## Setup
Install the dependencies with poetry
```
pip install poetry
poetry install
poetry shell
```
## Pytorch Lighthing
This script runs a `point cloud classifier` with several bakcbones on ModelNet under `100 lines`.
The bakcbones are `"kpconv", "pointnet2", "rsconv"`
```
poetry run python3 examples/pytorch-lightning/main.py kpconv
```
## FastAI
This script runs a `point cloud Unet segmentation model` with several bakcbones on Shapenet under `100 lines`.
The bakcbones are `"kpconv", "pointnet2", "rsconv"`
```
poetry run python3 examples/fastai/main.py rsconv
```