caffe cnn提取各层特征并可视化结果(Python接口)

亲测有用,之前参考的薛开宇的学习笔记,部分代码需要修改,自己修改后的版本如下:

</p>

#caffe特征的可视化
import numpy  as np
import matplotlib.pyplot as plt
import matplotlib.image as ima
import os
import pickle
import cv2

caffe_root='/home/falingling/caffe/'
import sys
sys.path.insert(0, caffe_root+'python')
import caffe

#显示的图表大小为 10,图形的插值是以最近为原则,图像颜色是灰色
plt.rcParams['figure.figsize'] = (10, 10)
plt.rcParams['image.interpolation'] = 'nearest'
plt.rcParams['image.cmap'] = 'gray'
#'examples/imagenet/caffe_reference_imagenet_model'
model_file = caffe_root+'models/bvlc_reference_caffenet/deploy.prototxt'  
pretrained = caffe_root+ 'models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel'
image_file = caffe_root+'examples/images/cat.jpg'  
#image_file = caffe_root+'examples/_temp/baby_74.jpg'
npload = caffe_root+ 'python/caffe/imagenet/ilsvrc_2012_mean.npy'  
#net=caffe.Classifier(caffe_root+'examples/imagenet/deploy.prototxt',caffe_root+'examples/imagenet/caffe_reference_imagenet_model')

net = caffe.Net(model_file,pretrained,caffe.TEST)
transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape})
transformer.set_transpose('data', (2,0,1))
transformer.set_mean('data',

  • 1
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
在使用PythonCaffe实现Bhattacharyya损失函数前,需要先了解Bhattacharyya距离和Bhattacharyya系数,它们是计算Bhattacharyya损失函数的基础。 Bhattacharyya距离是一种用于度量两个概率分布相似性的方法,其定义为: ![](https://cdn.jsdelivr.net/gh/1076827098/CDN/blog/nlp-chatbot/bhattacharyya_distance.png) 其中,P(x)和Q(x)分别为两个概率分布函数,x为概率变量。 Bhattacharyya系数是Bhattacharyya距离的指数形式,其定义为: ![](https://cdn.jsdelivr.net/gh/1076827098/CDN/blog/nlp-chatbot/bhattacharyya_coefficient.png) 在了解了Bhattacharyya距离和Bhattacharyya系数后,我们可以开始实现Bhattacharyya损失函数。下面是一个使用PythonCaffe实现Bhattacharyya损失函数的示例代码: ```python import caffe import numpy as np class BhattacharyyaLossLayer(caffe.Layer): def setup(self, bottom, top): if len(bottom) != 2: raise Exception("Need two inputs to compute Bhattacharyya loss.") # 检查输入数据维度是否匹配 if bottom[0].count != bottom[1].count: raise Exception("Inputs must have the same dimension.") self.diff = np.zeros_like(bottom[0].data, dtype=np.float32) self.epsilon = 1e-6 # 避免除数为0 def reshape(self, bottom, top): top[0].reshape(1) def forward(self, bottom, top): # 计算Bhattacharyya系数 self.diff[...] = bottom[0].data - bottom[1].data self.distance = np.sum(np.sqrt(np.abs(self.diff))) + self.epsilon self.bc = np.exp(-self.distance) # 计算Bhattacharyya损失 self.loss = -np.log(self.bc + self.epsilon) top[0].data[...] = self.loss def backward(self, top, propagate_down, bottom): if propagate_down: bottom[0].diff[...] = -(1 / (self.bc + self.epsilon)) * np.sign(self.diff) * np.exp(-self.distance / 2) / np.sqrt(np.abs(self.diff)) bottom[1].diff[...] = (1 / (self.bc + self.epsilon)) * np.sign(self.diff) * np.exp(-self.distance / 2) / np.sqrt(np.abs(self.diff)) ``` 在上面的代码中,我们定义了一个名为BhattacharyyaLossLayer的自定义层,实现了Bhattacharyya损失函数。在setup()函数中,我们首先检查输入的数据维度是否匹配,然后初始化diff和epsilon变量。在reshape()函数中,我们指定输出数据的维度。在forward()函数中,我们计算了Bhattacharyya系数和Bhattacharyya损失,并将损失值保存到top[0]中。在backward()函数中,我们计算了梯度,并将梯度值保存到bottom[0]和bottom[1]中。 需要注意的是,在计算梯度时,我们使用了符号函数和指数函数,这是由于Bhattacharyya距离的定义中包含了绝对值,导致其不可导。因此,我们使用了符号函数来代替导数的符号,使用指数函数来代替导数的大小。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值