我之前尝试有用青椒的代码来测试训练的模型,但是就各种报错,几经尝试,终于得到了一个跑的通的代码,其实就是小修小补一下就可以了,奈何是小白。。。
clc
clear all
net1 = dagnn.DagNN.loadobj(load('D:\exp\Sort_1000pics\net-deployed.mat'));
im = imread('D:\data\Sort_1000pics\test\300.jpg');
im_ = single(im);
im_ = imresize(im_, net1.meta.normalization.imageSize(1:2));
im_ = bsxfun(@minus, im_, net1.meta.normalization.averageImage);
net1.eval({'input',im_});
scores = net1.vars(net1.getVarIndex('prob')).value;
scores = squeeze(gather(scores));
[bestScore, best] = max(scores);
figure(1); clf; imagesc(im);
title(sprintf('%s (%d), score %.3f',...
net1.meta.classes.description{best}, best, bestScore));
修改到最后,其实也有点无语的感觉。