在 matlab 中使用 caffemodels

Question

这是我第一次尝试在 matlab 中使用 caffemodels，所以我有点迷茫 - 我从这个网站下载了一个 caffemodel (CNN)：https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/，我现在只是想使用它估计图像的年龄。到目前为止，我的尝试是这样的：

net = importCaffeNetwork('prot.prototxt', 'dex_imdb_wiki.caffemodel');

%% read image
image2 = imread('image2.jpg');
image2 = imresize(im2double(image2),[224 224]);

net(image2)

net 是一个具有 41 层的 CNN，其中第一层的输入大小为 [224 224 3]。当我运行 "net(image2)" 我得到错误 "Array indices must be positive integers or logical values".

而且我根本不知道如何解决这个问题..

非常感谢任何帮助。

Answer 1

NN 工具箱中的

classify 就是您要查找的内容。出于某种原因，您的 caffe 模型不适用于双 class 的图像。所以总的来说，以下是你需要的：

net = importCaffeNetwork('prot.prototxt', 'dex_imdb_wiki.caffemodel');
image2 = imread('image2.jpg');
image2 = imresize(image2, [224 224]);
age_pred = classify(net, image2);

在 matlab 中使用 caffemodels

using caffemodels in matlab

matlab

neural-network

caffe