在 Matlab 中提取具有可变批量大小的 Caffe 特征

Question

我知道如何使用随 Caffe 提供的 matcaffe_demo.m 提取 Caffe 专长/分数。但是，当使用此文件时，必须提供一个 prototxt 文件，该文件不仅可以确定网络架构，还可以确定输入维度，包括 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();