基于Point Transformers复现点云分割任务
https://arxiv.org/abs/2012.09164
使用连接下载 ShapeNet 数据集:下载地址
下载完成后解压到 data/shapenetcore_partanno_segmentation_benchmark_v0_normal
python train.py
Models | Accuracy | cat.mIOU | ins.mIOU |
---|---|---|---|
Point Transformer (paper) | None | 0.837 | 0.866 |
Point Transformer (ours) | 0.93456 | 0.79787 | 0.83578 |
感兴趣的同学关注我的公众号——可达鸭的深度学习教程:
AI Studio:https://aistudio.baidu.com/aistudio/personalcenter/thirdview/67156