经过不懈的努力,DF2NET终于在我的小MAC部署起来了。其中遇到了不少的问题,这里来记录一下。

github链接:​​https://github.com/xiaoxingzeng/DF2Net/​

1.环境要求

DF2NET三维模型部署_深度图

由于0.4.0版本的pytorch安装老报错,这里将版本修改为了0.4.1,使用pip进行安装。

DF2NET三维模型部署_python_02


对于Dlib和PIL,可以全部安装最新版本。

2.代码修改

DF2NET的demo.py文件处理的输入图片全部从img_list.txt文件中读取,而仔细观察可以发现img_list.txt的文件内容和文件目录中所包含的文件并不是一一匹配,缺少一部分文件,需要去寻找新的图片加入img目录或者是修改img_list.txt文件使其匹配。

DF2NET三维模型部署_深度学习_03

DF2NET三维模型部署_计算机视觉_04

3.版本差异导致的问题

在mac os中,macos不支持cuda,因此此处将num_workers 改为0,防止线程冲突,并将所有的cuda改为cpu进行运算。

DF2NET三维模型部署_github_05

同时,老版本的torch.autograd.Variable的使用方法产生了变更,在新的torch版本中,需要使用新的方法来使用Variable函数,否则会出现用户级异常。

DF2NET三维模型部署_深度图_06

运行结果

输入图片:

DF2NET三维模型部署_深度学习_07

使用在线matlab进行模型深度图的查看和预览。

DF2NET三维模型部署_计算机视觉_08


得到深度图。

DF2NET三维模型部署_计算机视觉_09


通过运行细节图来生成最终的三维模型。

DF2NET三维模型部署_计算机视觉_10