matlab - Matlab で可変バッチ サイズを使用して Caffe 機能を抽出する

Question

matcaffe_demo.mCaffeと共に提供されている を使用して、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();