20201222

配置环境

tensorflow-gpu 1.10.0
h5py
opencv-python
CUDA Version 10.0.130
此处要注意配置环境/usr/local/cuda-10.1/bin/nvcc下需要有nvcc文件,否则后面编译会不通过

点云可视化软件

cloudcompare
Meshlab

克隆项目

git clone https://github.com.cnpmjs.org/yongbai123/pointnet2_python3.git
将下载的域名github.com改为github.com.cnpmjs.org可以加速

编译tf-ops和utils

修改对应文件夹下sh文件中的cuda版本,编译生成so.so文件,可用ls -al查看

数据集

data/modelnet40_normal_resampled
data/modelnet40_ply_hdf5_2048(无法向量)

训练modelnet40

python train.py
tensorboard --logdir log
默认训练参数:batch_size=16, decay_rate=0.7, decay_step=200000, gpu=0, learning_rate=0.001,
log_dir=‘log’, max_epoch=251, model=‘pointnet2_cls_ssg’, momentum=0.9, normal=False,
num_point=1024, optimizer=‘adam’

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 酷酷鲨 设计师:CSDN官方博客 返回首页