pytorch学习
键盘强者
这个作者很懒,什么都没留下…
展开
-
记使用Pytorch的hook机制提取特征时踩的一个坑
因为项目需求,需要用DenseNet模型提取图片特征,在使用Pytorch的hook机制提取特征,调试的时候发现提取出来的特征数值上全部大于等于0。很明显提取出来的特征是经过ReLU的。现在来看一下笔者是怎么定义hook的:...原创 2020-09-21 11:41:26 · 1199 阅读 · 2 评论 -
Pytorch实现Grad-CAM
因为有一个分类模型始终得不到比较好的结果,所以查阅资料发现可以使用类激活图(CAM)的方式来看看模型究竟学到了什么,是不是重点关注我们希望的区域,所以特此记录一下学习过程。简介CAM全称是Class Activation Map,即类激活图。可以理解为模型某次预测中,对哪个区域的响应最大,可以说这个区域很大程度上决定了模型这次预测结果。以猫狗大战为例,如下面这张图一个任意的分类网络,输入一个既包含着一只狗,又包含着一只猫的猫狗合影图片,它会输出一个2维度的概率向量,分别对应着图片分类为猫和图片分类原创 2020-08-11 20:47:11 · 6892 阅读 · 1 评论 -
加载pytorch预训练模型densenet121出现Missing key(s) in state_dict Unexpected key(s) in state_dict
在手动加载densenet121权重时,报了一下的错误:原创 2020-08-04 18:09:46 · 1983 阅读 · 0 评论