loicland
commented
1 year ago Hi!
We are releasing a new version of SuperPoint Graph called SuperPoint Transformer (SPT).
https://github.com/drprojects/superpoint_transformer
It is better in any way:
| ✨ SPT in numbers ✨ |
|---|
| 📊 SOTA results: 76.0 mIoU S3DIS 6-Fold, 63.5 mIoU on KITTI-360 Val, 79.6 mIoU on DALES |
| 🦋 212k parameters only! |
| ⚡ Trains on S3DIS in 3h on 1 GPU |
| ⚡ Preprocessing is x7 faster than SPG! |
| 🚀 Easy install (no more boost!) |
If you are interested in lightweight, high-performance 3D deep learning, you should check it out. In the meantime, we will finally retire SPG and stop maintaining this repo.
when I run for FOLD in 1 2 3 4 5 6; do \ CUDA_VISIBLE_DEVICES=0 python learning/main.py --dataset s3dis --S3DIS_PATH /home/lanson/桌面/Code/superpoint_graph/S3DIS/data/ --cvfold $FOLD --epochs 350 --lr_steps '[275,320]' \ --test_nth_epoch 50 --model_config 'gru_10_0,f_13' --ptn_nfeat_stn 14 --nworkers 2 --odir "results/s3dis/best/cv${FOLD}"; \ done
it shows that: