CNN可视化工具:用于深度卷积神经网络中的单元可视化的工具包
介绍
该存储库包含用于可视化深CNN内的单元(或称为神经元或卷积过滤器)的代码和结果。 一些代码已用于深度场景CNN中出现的物体检测器》。 您可以将此工具包与天真的一起使用,并可以将其编译为matcaffe和pycaffe或 。 该工具包包含以下功能:
(新!)PyTorch脚本:
pytorch_extract_feature.py :用于为任何给定图像在CNN模型的选定层提取CNN特征的代码。
pytorch_generate_unitsegments.py :用于生成所选层上所有单元的可视化的代码。
Matlab脚本:
extract_features.m :用于为任何给定图像提取所选层上的CNN激活的代码。
generate_unitsegments.m :用于生成选定层中所有单元的可视化的代码。
unit
1