PyTorch/[PyTorch 学习笔记] 5.2 Hook 函数与 CAM 算法
本章代码:
这篇文章主要介绍了如何使用 Hook 函数提取网络中的特征图进行可视化,和 CAM(class activation map, 类激活图)
Hook 函数概念
Hook 函数是在不改变主体的情况下,实现额外功能。由于 PyTorch 是基于动态图实现的,因此在一次迭代运算结束后,一些中间变量如非叶子节点的梯度和特征图,会被释放掉。在这种情况下想要提取和记录这些中间变量,就需要使用 Hook 函数。
PyTorch 提供了 4 种 Hook 函数。
![PyTorch/[PyTorch 学习笔记] 5.2 Hook 函数与 CAM 算法](https://image.zhangxiann.com/andrea-hagenhoff-dIXCt2zUzV0-unsplash.jpg)

![PyTorch/[PyTorch 学习笔记] 3.1 模型创建步骤与 nn.Module](https://image.zhangxiann.com/mahdi-soheili-cOdXOk0UT8w-unsplash.jpg)
![PyTorch/[PyTorch 学习笔记] 2.3 二十二种 transforms 图片数据预处理方法](https://image.zhangxiann.com/markus-spiske-9R72Kd2oTGg-unsplash.jpg)



