【环境搭建】CVP-MVSNet实验配置及训练测试

环境配置

需要下载作者提供的数据集,跟MVSNet的数据集不是很一样,但都不大2G+1G

整体环境推荐不要复用MVSNet,主要是PyTorch版本比较新,再依照requirements安装下没安过的库即可

如果opencv-python在conda中无法安装成功,就直接用pip安装,一样可以用的

我最开始复用了MVSNet的cuda和torch环境,会在很多位置报错RuntimeError: output with device cuda:0 and dtype Float doesn't match the desired device cuda:0 and dtype Double,定位到报错位置将变量改为xx.double()进行格式转换会消除一些,但有些始终无法修改成功,所幸从头安装了

Train

修改下train.sh里的配置,直接运行就可以,对GPU的消耗不是很大,迭代的速度也比较快

Test / Eval

安排好数据集的位置就可以直接运行sh eval.sh进行测试即可,最终深度图结果会在outputs_pretrained

一点需要注意的是,如果运行时报错则需要指定一个GPU: CUDA_VISIBLE_DEVICES=0

eval之后会在outputs中生成每个场景的confidencedepth_est

fusion

记得要使用python2进行fusion,我最后是重新安装了一个python2的conda环境成功的

安装好环境,链接一下fusibile的位置到CVP-MVSNet/fusibile/目录下

ln -s /media/public/yan1/doublez/realdoubleZ/Developer/MVS/fusibile/build/fusibile fusion/fusibile

fusion之后会在outputs里的fusibile_fused生成每个场景的文件夹,文件夹里就是那些一大堆内容,最核心的是

  • cams/
  • images/
  • consistencyCheck/final3d_model.ply
  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 9
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

doubleZ0108

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值