ppogg/YOLOv5-Lite

heatmap功能报错

koushita opened this issue · 2 comments

keyerro
print完模型之后有21层,但是定位的是23,有大佬能帮帮忙吗

请问是哪个模型呢

感谢大佬回复。
是v5lite-s。
目前问题已解决,把main.py里的target-layer默认值改为default='model_17_cv3_act',可以生成heatmap。