三维重建工具001-——正确安装kaolin库并在SMR上运行

三维重建工具001-——正确安装kaolin库并在SMR上运行

三维重建工具001-正确安装kaolin库并在SMR上运行

---

文章目录

• 1 为什么kaolin呢
• 2 kaolin踩的坑
• 3 具体安装 步骤
• 4 结束

如果时间及任务重，直接点目录3，因为其他全不能解决问题。 1 为什么kaolin呢

   kaolin由NViDIA开发的一套用于加速 3D 深度学习研究的工具。在阅读CVPR论文 SMRSelf-Supervised 3D Mesh Reconstruction From Single Images，被作者的一些新颖的思路所处触动，决定和小伙伴一起复现该论文，采用的开源地址是kaolin。在安装该库的过程中遇到一些问题，所以记录下来，希望能减少踩坑的过程。

2 kaolin踩的坑

  首先我们的系统是ubuntu18.04，参考 官网链接和kaolin链接进行安装时，会对python，torch， cython,scipy,pillow ，tqdm和usd-core进行安装，安装对应版本进行安装，我们在18。唯一一个可能会出错的就是USD-core，会发生无法下载的问题。两种方法解决一种到https://pypi.org/search/?q=usd-core到该网址直接搜对应版本下载并安装，另一种方法是在setup.py运行到这个下载usd-core卡住的时候，把链接拷贝下来，自己主动下载并安装。
如果没安装好usd-core会报这样一个错误
  注意，在这种情况下，由于卡死在下载usd-core部分会导致有另外两个包没有下载成功。

提示把他们安装玩就好。
完全只按照官网添加以下两个链接的话还是会出现编译错误
export IGNORE_TORCH_VER=1
export KAOLIN_INSTALL_EXPERIMENTAL=1

注意重点解决办法就在kaolin的链接里，需要继续添加
export USE_CYTHON=1
然后用env进行查询

3 具体安装 步骤

1）创建环境

conda create -n lzq_kaolin1 python==3.6.2 #我们创建的是基于python3.6.2
conda activate lzq_kaolin1   #conda激活环境

2）下载kaolin，有时候下载库可能遇到下载问题，可以改到国内的gitee这样的代码平台进行clone，另外一个方式就是手动下载。

git clone --recursive https://github.com/NVIDIAGameWorks/kaolin   #--recursive是循环下载的意思，因为kaolin库用到了其他库

3）环境变量添加，这一步很重要。

export USE_CYTHON=1   #在cuda上编译的必要工具
export IGNORE_TORCH_VER=1
export KAOLIN_INSTALL_EXPERIMENTAL=1
TORCH_CUDA_ARCH_LIST=6.1 7.0  #这一步我们后面有两块算力不一样的显卡，自己的要去NVIDIA官网 确认下自己显卡算力。

下看确认路径是否加入并激活新的环境

env     #linux 命令
source ~/.bashrc

4）依赖包安装

conda install cython==0.29.20
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch
conda install scipy==1.2.0
conda install tqdm==4.61.0
conda install flask==1.1.2
pip install usd_core-21.11-cp36-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl 
conda install tornado==6.0.4

5）安装kaolin

cd kaolin  #进入kaolin的安装路径
git checkout v0.9.1  #下载的kaolin是一个库，里面包含很多版本，用该命令切换到0.9.1版本这个分支，这个是git版本管理工具的使用
python setup.py develop   #直接运行代码，进行环境的编译和一些注册功能

到这一步基本就安装完成。想验证以下的话继续看6），没时间的话，以下就不重要了。
6）验证kaolin的安装问题

python -c "import kaolin; print(kaolin.__version__)"  #这步骤一般没问题，结果会打印一个版本号
pip install -r tools/ci_requirements.txt #安装一些测试需要的环境
pytest tests/python/       #测试

注意在测试的时候会报一些错误，先不要慌，等测试完看问题报告。


看到这里先不要慌，因为在kaolin出现finished processing 后就可以认为安装好了，但是爆了这些错误，去找一下原因就好，28个faild和404errors其实都是安装ci_requirements.txt文件依赖的时候，里面没有下载数据集。
没有下载数据集，所有会报一系列的no found errors，由于没有数据所以len>0不成立又会发生failed问题。

4 结束

  这里，我们遇到的问题基本就已经都分享了。安装玩kaolin之后去复现SMR基本没有问题了，因为一方面论文的结构简单，另一方面借助lkaolin这样的库来处理三维问题，所以代码就很简洁了。
  最后最后就是参考的链接了。
worldlink kaolin
kaolin官网
SMR github
SMR PDF百度网盘 密码:8to3