画saliency intensity map图,pytorch版本

最新推荐文章于 2024-02-17 22:11:57 发布

qiaoqiao2332

最新推荐文章于 2024-02-17 22:11:57 发布

阅读量1.5k

点赞数 2

分类专栏： pytorch 文章标签： pytorch 可视化

本文链接：https://blog.youkuaiyun.com/u011668104/article/details/108288694

版权

本文记录了使用PyTorch实现saliency intensity map过程中遇到的问题及解决方案，特别是解决grad为None的问题，需要添加hook。通过示例展示了模型检测AU12的情况，分析了结果中嘴角和眼周亮点的可能原因。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

之前写过一篇caffe版本的，这次需要pytorch版本的发现踩坑了，主要踩坑是grad不管怎么弄都是None，查了资料发现是pytorch释放了，所以需要加hook才行。记录一下备忘。

import cv2
import torch

img_path = 'path/test_img.jpg' #测试用的图片路径
img_and_saliency_map_name = 'outpath/img_and_saliency_map.jpg'#待求的 图片+saliency intensity map的叠加图路径

#这里按照你自己的网络测试过程把图片读进去就行，反正输入前一般是[N,C,H,W],我这是一张3通道彩色图，176*176的，注意requires_grad=True
img = get_img(img_path)
img = torch.autograd.Variable(img, requires_grad=True)
img = img.view((1,img.size()[0],img.size()