三维重建工具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会报这样一个错误三维重建工具001-——正确安装kaolin库并在SMR上运行_第1张图片
  注意,在这种情况下,由于卡死在下载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

三维重建工具001-——正确安装kaolin库并在SMR上运行_第2张图片

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/       #测试

注意在测试的时候会报一些错误,先不要慌,等测试完看问题报告。
三维重建工具001-——正确安装kaolin库并在SMR上运行_第3张图片
在这里插入图片描述
看到这里先不要慌,因为在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

你可能感兴趣的:(三维重建,深度学习,pytorch,python)