深度学习算法的点云分割-PointNet（五）

转自:https://blog.csdn.net/u012986684/article/details/78261165

大家好,今天给大家介绍下cvpr2017 一篇文章Pointnet 语义分割,该网络基于tensorflow 写的,非常轻巧方便,但是文章和代码有一定出入,在训练过程中出现过拟合现象,大概训练了10个小时多.

1    需要准备的东东

(1) Ubuntu 16.04  GTX1060 6G

(2) CUDA 安装

(3) tensorflow 安装 

2 下载源码

git clone https://github.com/charlesq34/pointnet.git

3 网络结构 (使用tensorBoard 查看网络graph)

4 训练网络

查看 训练情况:

6 查看下分割情况

总结:

par_seg 做了部分语义分割, s_seg 场景的分割,写个test.py 就可以了