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

4 次查看（过去 30 天）
Ryoya Takahashi
Ryoya Takahashi 2017-8-3
回答： Tohru Kikawada 2017-8-3
現在、VGG16を利用して転移学習を行い画像判別を行おうとしております。 現状はVGG16の読み込み→layerの編集→訓練用画像の学習→テスト画像の入力→性能評価の段階で実施しておりますが、 VGG16の読み込み→layerの編集→訓練用画像の学習までを実行し、出来上がったNeural Networkを保存する方法はありますでしょうか？ また、保存したこのNeural Networkの呼び出し方法もお教えいただければと思います。

请先登录，再回答此问题。

采纳的回答

Tohru Kikawada
Tohru Kikawada 2017-8-3
MAT-fileに保存ではだめでしょうか。 save コマンドで保存できます。
netTransfer = trainNetwork(trainDigitData,layers,optionsTransfer);
save('netTransfer.mat','netTransfer');
load コマンドで読み込みもできます。
d = load('netTransfer.mat');
netTransfer = d.netTransfer;
YPred = classify(netTransfer,testDigitData);

更多回答（1 个）

michio
michio 2017-8-3
matファイルに保存しておくことが可能です。例えば訓練用画像の学習を
convnet = trainNetwork(trainDigitData,layers,options);
の様に実施した場合、
save('trainedNetwork.mat', 'convnet');
と trainedNetwork.mat に保存します。再び読み込む場合は、例えば
load('trainedNetwork.mat','convnet')
と実行すれば convnet 変数が読み込まれその後の処理に使用できるかと思いますが、いかがでしょうか？

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!