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

Question

matcaffe_demo.m我知道如何使用与 Caffe 一起提供的来提取 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();