第六章 PyTorch 可视化模块
第六章简介
本章介绍可视化工具,包括TensorBoard可视化工具,混淆矩阵,CNN卷积核与特征图可视化,分类模型注意力算法——Grad CAM,模型参数量可视化。
首先对强大的可视化工具TensorBoard进行讲解,介绍其提供的十多个数据可视化API。
然后借助Tensorboard的绘图功能,观察CNN卷积核与特征图的变化过程,同时对分类混淆矩阵进行分析。
接着介绍一个实用的分类模型注意力机制可视化算法——Grad CAM。
最后介绍一系列对模型参数量分析的工具。