Training後の​NeuralNetw​orkの保存と呼び出​し方法について

現在、VGG16を利用して転移学習を行い画像判別を行おうとしております。 現状はVGG16の読み込み→layerの編集→訓練用画像の学習→テスト画像の入力→性能評価の段階で実施しておりますが、 VGG16の読み込み→layerの編集→訓練用画像の学習までを実行し、出来上がったNeural Networkを保存する方法はありますでしょうか？ また、保存したこのNeural Networkの呼び出し方法もお教えいただければと思います。

Answer 1

[Tohru Kikawada]

MAT-fileに保存ではだめでしょうか。 save コマンドで保存できます。

netTransfer = trainNetwork(trainDigitData,layers,optionsTransfer);
save('netTransfer.mat','netTransfer');

load コマンドで読み込みもできます。

d = load('netTransfer.mat');
netTransfer = d.netTransfer;
YPred = classify(netTransfer,testDigitData);

Answer 2

[michio]

matファイルに保存しておくことが可能です。例えば訓練用画像の学習を

 convnet = trainNetwork(trainDigitData,layers,options);

の様に実施した場合、

 save('trainedNetwork.mat', 'convnet');

と trainedNetwork.mat に保存します。再び読み込む場合は、例えば

 load('trainedNetwork.mat','convnet')

と実行すれば convnet 変数が読み込まれその後の処理に使用できるかと思いますが、いかがでしょうか？