caffe matlab 提取全链接层特zheng

最新推荐文章于 2023-05-22 19:58:07 发布

qq_2773878606

最新推荐文章于 2023-05-22 19:58:07 发布

阅读量1.8k

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/qq_18343569/article/details/56028312

版权

深度学习专栏收录该内容

2 篇文章 0 订阅

订阅专栏

clc;
clear all;
addpath('..') %加入+caffe路径
load da2.mat
%%
% caffe.set_mode_cpu();%设置CPU模式
model = 'deploy.prototxt';%模型
weights = 'lenet_iter_2066.caffemodel';%参数

net=caffe.Net(model,'test');%测试
net.copy_from(weights); %得到训练好的权重参数
IMAGE_SIZE = [60, 36];
net.blobs('data').reshape([ 60 36 1 1]);
m=size(data);
feature2=[];
for i=1:m(3)
image = data(:,:,i);
image=imresize(image,IMAGE_SIZE);
paramer=cell(2);

image=single(image);
headpose=single(headpose);
paramer{1}=image;
paramer{2}=headpose(:,i);
out = net.forward(paramer);
 result = net.blobs('cat').get_data();
 feature2=[feature2 result];
 
end