-
Notifications
You must be signed in to change notification settings - Fork 144
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
关于如何使用提供的权重推理? #15
Comments
@Zhutianyi7230 我也使用作者提供的模型进行推理,但是如何对推理结果进行可视化,请问你有进展吗? |
@cyn-liu 我暂时还没有推理,按作者所说推理和可视化代码应该都在path/to/MapTR/tools/maptr中,我还不知道如何使用,如果您能跑通推理流程的话可以告诉我一下大致如何操作吗,谢谢! |
@Zhutianyi7230 推理作者在这里写了说明:https://github.com/hustvl/MapTR/blob/main/docs/train_eval.md |
@cyn-liu 我认为./tools/dist_test_map.sh这个应该是在数据集上验证用的。想要用权重直接推理的话可以看一下/tools/maptr/vis_pred.py这个代码,我还没仔细看过,但是它里面有plt函数,按该代码要求设置参数,应该可以可视化。 |
@Zhutianyi7230 推理需要数据,用数据集的数据并没有什么问题,问题是使用dist_test_map.sh推理后生成的文件如何可视化呢? 可能使用vis_pred.py也可以推理,但是我在这个文件中并没有看到如何设置推理数据的地方。 |
@Zhutianyi7230 我使用/tool/maptr/vis_pred.py这个脚本可视化推理结果是成功了的,这个脚本可视化的是整个测试集。 |
可视化脚本使用: python ./tool/maptr/vis_pred.py /path/to/config /path/to/ckpt --gt-format fixed_num_pts polyline_pts --score-thresh 0.4。摆放位置就follow教程就好,可视化的脚本不需要专门设置路径,通过读取config会自动设置好这些 |
感谢作者开源这一非常优秀的工作!
我想要使用您提供的训练好的权重进行推理,复现一个demo,
也就是
path/to/MapTR/tools/maptr
中的一些程序(vis_pred.py等),请问具体该如何使用呢(用什么命令参数,各个文件的摆放位置等)?感谢您的交流!
The text was updated successfully, but these errors were encountered: