XiandaGuo / OpenStereo

OpenStereo: A Comprehensive Benchmark for Stereo Matching and Strong Baseline
423 stars 49 forks source link

无法推理 #118

Closed Hlll-huang closed 4 weeks ago

Hlll-huang commented 3 months ago

你好,我想请问,推理lightstereo时候,模型的权重应该设置在哪里?为什么直接使用lightstereo还需要下载moblenet的权重???

CaptainRui1000 commented 3 months ago

你好,我想请问,推理lightstereo时候,模型的权重应该设置在哪里?为什么直接使用lightstereo还需要下载moblenet的权重???

OpenStereo/stereo/modeling/models/lightstereo/backbone.py Backbone里面的pretrained改成=False就不需要下载moblenet了

lightstereo的权重可以在model zoo找到,https://drive.google.com/drive/folders/1eGtN_JAKy684ZP8gLTUemSgQxZir8hLf

Hlll-huang commented 3 months ago

已解决,感谢,但又存在新的问题,尝试使用官方权重推理kitti的第一张数据集,但得到的效果很差,想问问是什么原因,有解决方案吗? 000000_10

Hlll-huang commented 3 months ago

伪彩色图,如下所示 0

CaptainRui1000 commented 3 months ago

已解决,感谢,但又存在新的问题,尝试使用官方权重推理kitti的第一张数据集,但得到的效果很差,想问问是什么原因,有解决方案吗? 000000_10

请问是怎样的推理流程?

Hlll-huang commented 3 months ago

已解决,感谢,但又存在新的问题,尝试使用官方权重推理kitti的第一张数据集,但得到的效果很差,想问问是什么原因,有解决方案吗? 000000_10

请问是怎样的推理流程? 我在cfg的lightstereo_s_kitti.yaml文件里面的参数PRETRAINED_MODEL添加了官网的训练权重,然后运行python tools/infer.py 添加cfg_file和左右图路径。 1