search

ICSResearch / TrapAttention

MIT License
10 stars 0 forks source link

感谢作者的优秀工作，我在运行test.py的时候遇到了一个问题，就是from .trap1 import TrappedHead1 #2

Open WilliamOChai opened 6 months ago

WilliamOChai commented 6 months ago

感谢作者的优秀工作，我在运行test.py的时候遇到了一个问题，显示No module named 'depth.models.decode_heads.trap1'，我在这个py里面没有发现这个类。 此外，作者能否提供single inference、 train、test相关的命令，或者使用方法！ 非常感谢！

2J472no commented 5 months ago

感谢关注。trap1是消融实验遗留下来的，直接删除就行了。关于train和test可以参考https://github.com/zhyever/Monocular-Depth-Estimation-Toolbox 。 我们使用这个仓库的训练和测试的框架。