参考网址:Github网址
环境配置与程序跑通
- 环境:TF1.2 GPU + Python 2.7(3也行)+ Ubuntu 14.04 + 依赖库(cv2、h5等);
- 下载scannet数据集放到data文件夹下;在train.py中修改关于模型和数据集位置的相关代码;
- 针对tf_op文件夹:
注:
pip install tensorflow-gpu==版本号
pip install opencv-python (只用于python环境)
sudo pip install cython
sudo apt-get install libhdf5-dev
sudo pip install h5py
(或者 python3 -m pip install h5py )
- 在终端python环境下输入上图中阴影部分的代码,根据当前tensorflow的版本,修改.sh文件,如下图:
- sh该文件夹下所有的.sh文件
- 举荔枝:
- cd tf_utils/tf_ops/sampling
chmod +x tf_sampling_compile.sh
./tf_sampling_compile.sh
- cd tf_utils/tf_ops/sampling
- 举荔枝:
END!!!