(DeepLearning Tensorflow) PointNet

• PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation

特に何の苦労も無く動いた

インストール

python2.7という指示なので仮想環境を作る。

conda create -n pointnet python=2.7 h5py numpy scipy matplotlib tensorflow-gpu jupyter
source activate pointnet
git clone https://github.com/charlesq34/pointnet.git

特にビルド無し

テスト

データセットも勝手に準備してくれる

cd pointnet
python train.py

でtrainが始まる。半日くらい待つ(1080)。

python evaluate.py --visu

でdumpディレクトリに失敗例を書き出してくれる。ファイル名を見ると何をどう誤検知したかわかる。

label: desk, pred: piano

label: bookshelf, pred: bottle

label: plant, pred: flower_pot