基于Matlab的变批量Caffe特征提取

Question

我知道如何使用与Caffe一起提供的matcaffe_demo.m来提取Caffe专长/分数。但是，在使用该文件时，必须提供一个原型文件，该文件不仅确定网络体系结构，而且还确定包括batch_size在内的输入维度。

由于我正在处理可变序列长度的视频帧，我需要一种方法来使用matcaffe_demo.m以及可变的批处理大小。

有人知道怎么做吗？

它可能需要从matcaffe_demo.m中更改这一行。

% Initialize a network
net = caffe.Net(net_model, net_weights, phase);

对于动态传递当前批处理大小的对象，所需的批大小动态地传递给caffe。

Answer 1

最后我使用了reshape函数：

net = caffe.Net(net_model, net_weights, phase);
net.blobs('data').reshape([dim1 dim2 numChannels numFrames]);
scores = net.forward(inputData);
caffe.reset_all();