2-23年3月20日,
yolov5 7.0热力图实验,加入了shufflenetv2 conv5 - ..yaml文件
用自己的权重生成热力图。遇到bug
解决方法:
打开模型的yaml文件。找到detect的层数
然后在main_gradcam.py文件中target_layers=['model_15_cv3_act', 'model_18_cv3_act', 'model_21_cv3_act']
这三个数字改成自己的网络结构。
成功
2-23年3月20日,
yolov5 7.0热力图实验,加入了shufflenetv2 conv5 - ..yaml文件
用自己的权重生成热力图。遇到bug
解决方法:
打开模型的yaml文件。找到detect的层数
然后在main_gradcam.py文件中target_layers=['model_15_cv3_act', 'model_18_cv3_act', 'model_21_cv3_act']
这三个数字改成自己的网络结构。
成功