SPoTr

KO | EN

gitlite — search

#javascript #ai #typescript #llm #go #golang #android #machine-learning #rust #linux #java #awesome #cli #pytorch

(★ 113)

Official pytorch implementation of "Self-positioning Point-based Transformer for Point Cloud Understanding" (CVPR 2023).

SPoTr 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Self-positioning Point-based Transformer for Point Cloud Understanding

Official pytorch implementation of "[Self-positioning Point-based Transformer for Point Cloud Understanding](https://openaccess.thecvf.com/content/CVPR2023/html/Park_Self-Positioning_Point-Based_Transformer_for_Point_Cloud_Understanding_CVPR_2023_paper.html)" (CVPR 2023).

![Figure](https://raw.githubusercontent.com/mlvlab/SPoTr/main/assets/main_figure.png)

> Jinyoung Park 1* , Sanghyeok Lee 1* , Sihyeon Kim 1 , Yunyang Xiong 2 , Hyunwoo J. Kim 1† 
> 1 Korea University, 2 Meta Reality Labs

## Setup
- Clone repository 
```
git clone https://github.com/mlvlab/SPoTr.git
cd SPoTr
```
- Install packages with a setup file
```
bash install.sh
```
- Dataset

```
mkdir -p data/S3DIS/
cd data/S3DIS
gdown https://drive.google.com/uc?id=1MX3ZCnwqyRztG1vFRiHkKTz68ZJeHS4Y
tar -xvf s3disfull.tar
cd ../../

cd data && mkdir ShapeNetPart && cd ShapeNetPart
gdown https://drive.google.com/uc?id=1W3SEE-dY1sxvlECcOwWSDYemwHEUbJIS
tar -xvf shapenetcore_partanno_segmentation_benchmark_v0_normal.tar
cd ../../
```

## Run Experiments
If you want pretrained models, you can download pretrained models via this [URL](https://drive.google.com/drive/folders/1s3mfjMG6cRfH9ftDwaukTWLsi0f_eBGy?usp=sharing)
### SNPart
- Train
```
CUDA_VISIBLE_DEVICES='0' python examples/shapenetpart/main.py --cfg cfgs/shapenetpart/spotr.yaml
```
- Inference
```
CUDA_VISIBLE_DEVICES='0' python examples/shapenetpart/main.py --cfg cfgs/shapenetpart/spotr.yaml mode=test --pretrained_path ckpt/ShapeNetPart/ckpt_best.pth
```
### S3DIS
- Train
```
CUDA_VISIBLE_DEVICES='0' python examples/segmentation/main.py  --cfg cfgs/s3dis/spotr.yaml
```
- Inference
```
CUDA_VISIBLE_DEVICES='0' python examples/segmentation/main.py  --cfg cfgs/s3dis/spotr.yaml  mode=test --pretrained_path ckpt/S3DIS/ckpt_best.pth
```

## Acknowledgement
This repo is built upon OpenPoints.
```
https://github.com/guochengqian/openpoints
```

SPoTr

SPoTr 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

SPoTr

SPoTr 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README