这篇博客教大家,如何对网络实现可视化。无论你是哪种CNN网络均适用这种方法,非常好用!!!学会这种方法十分钟搞定任何网络可视化,如果遇到问题欢迎评论区留言。
下面先说必要加的模块,后进行四组最新分类CNN网络实例演示(ConvNeXtV2、EfficientNet、MnasNet、VGNet),最后列举遇到的问题(最新代码中已做修改按照文章中的跑就行)
1. 模块(从上至下)
# 添加6---定义辅助函数,获取指定层名称的特征,定义钩子hook
activation = {
}
def get_activation(name):
def hook(model,input,output):
activation[name]