XiandaGuo / OpenStereo

OpenStereo: A Comprehensive Benchmark for Stereo Matching and Strong Baseline
369 stars 36 forks source link

无法推理 #118

Open Hlll-huang opened 1 month ago

Hlll-huang commented 1 month ago

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

CaptainRui1000 commented 1 month 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 1 month ago

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

Hlll-huang commented 1 month ago

伪彩色图,如下所示 0

CaptainRui1000 commented 1 month ago

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

请问是怎样的推理流程?

Hlll-huang commented 1 month ago

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

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

rachitagarwal2603 commented 22 hours ago

In the tools/infer.py, on giving left and right image as input of dimension HxW. I see model predicts Hx(Wx3) dimension output. Here, as per my understanding, first channel represents disparity map. What does the other 2 channels represent?