将filter或者activation保存成图片。
Posted 机器学习的小学生
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了将filter或者activation保存成图片。相关的知识,希望对你有一定的参考价值。
在python代码中,将滤波器filter保存为.mat文件(这里是为了利用matlab写图片,当然也可以在python中直接写图片):
filters = net.params['conv1'][0].data
filters = (filters - filters.min()) / (filters.max() - filters.min())
vis_square(filters.transpose(0, 2, 3, 1))
sio.savemat(snap_path + 'filter.mat','filter':filters)
plt.show()
利用matlab写图片:
close all;
filter = load('filter.mat');
filter = filter.filter;
filter = permute(filter,[1 3 4 2]);
disp(size(filter));
for i = 1:size(filter,1)
img = filter(i,:,:,:);
img = squeeze(img);
img1=imresize(img,10);
subplot(1, 2, 1)
imshow(img1,'Border','tight');
saveas(gcf,['test.jpg']);
img0=uint8(img*255);%这样可以,这个快一点
%img0=im2uint8(img0);%这样也可以,这个很慢
img0=imresize(img0,10);
imwrite(img0,['test.png']);
subplot(1, 2, 2)
imshow(img0,'Border','tight');
%pause(1);
break;
end
以上是关于将filter或者activation保存成图片。的主要内容,如果未能解决你的问题,请参考以下文章