经过不懈的努力,DF2NET终于在我的小MAC部署起来了。其中遇到了不少的问题,这里来记录一下。
github链接:https://github.com/xiaoxingzeng/DF2Net/
1.环境要求
由于0.4.0版本的pytorch安装老报错,这里将版本修改为了0.4.1,使用pip进行安装。
对于Dlib和PIL,可以全部安装最新版本。
2.代码修改
DF2NET的demo.py文件处理的输入图片全部从img_list.txt文件中读取,而仔细观察可以发现img_list.txt的文件内容和文件目录中所包含的文件并不是一一匹配,缺少一部分文件,需要去寻找新的图片加入img目录或者是修改img_list.txt文件使其匹配。
、
3.版本差异导致的问题
在mac os中,macos不支持cuda,因此此处将num_workers 改为0,防止线程冲突,并将所有的cuda改为cpu进行运算。
同时,老版本的torch.autograd.Variable的使用方法产生了变更,在新的torch版本中,需要使用新的方法来使用Variable函数,否则会出现用户级异常。
运行结果
输入图片:
使用在线matlab进行模型深度图的查看和预览。
得到深度图。
通过运行细节图来生成最终的三维模型。