Skip to content

基于Pytorch复现Point-Transformer,用于ShapeNet数据集点云分割

License

Notifications You must be signed in to change notification settings

daqingyi770923/Point-Transformer-Pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch Implementation of Point Transformer

基于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

关注我的公众号:

感兴趣的同学关注我的公众号——可达鸭的深度学习教程:

在这里插入图片描述

联系作者:

B站:https://space.bilibili.com/470550823

CSDN:https://blog.csdn.net/weixin_44936889

AI Studio:https://aistudio.baidu.com/aistudio/personalcenter/thirdview/67156

Github:https://github.com/Sharpiless

Reference:

https://github.com/qq456cvb/Point-Transformers

About

基于Pytorch复现Point-Transformer,用于ShapeNet数据集点云分割

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%