深度學習算法的點雲分割-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 就可以了