在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();