データの読み込み

新しいイメージを解凍してイメージデータストアとして読み込みます。この非常に小さいデータセットには75枚のイメージのみが含まれています。データを学習データセットと検証データセットに分割します。イメージの70%を学習に使用し,30%を検証に使用します。

解压(“MerchData.zip”）;imd = imageDatastore (“水星”,...“IncludeSubfolders”是的，...“LabelSource”,“foldernames”)；[imdsTrain，imdsValidation]=splitEachLabel（imds，0.7）；

事前学習済みのネットワークの読み込み

事前学習済みのGoogLeNetネットワークを読み込みます。深度学习工具箱™模型谷歌网サポート パッケージがインストールされていない場合、ダウンロード用リンクが表示されます。

別の事前学習済みネットワークを試すには,この例をMATLAB®で開き,別のネットワークを選択します。たとえば,googlenetよりも高速なネットワークである挤压网を試すことができます。この例は,他の事前学習済みネットワークを使用して実行することもできます。使用可能なすべてのネットワークについては,事前学習済みのネットワークの読み込みを参照してください。

网=googlenet;

analyzeNetworkを使用して、ネットワーク アーキテクチャを対話的に可視化し、ネットワーク層についての詳細情報を表示します。

analyzeNetwork(净)

ネットワークの层プロパティの最初の要素はイメージ入力層です。水壶ネットワークの場合、この層にはサイズが 224x224x3の入力イメージが必要です。ここで、3.はカラー チャネルの数です。その他のネットワークには異なるサイズの入力イメージが必要な場合があります。たとえば、例外ネットワークにはサイズ 299 x 299 x 3のイメージが必要です。

net.Layers（1）

ans = ImageInputLayer with properties: Name: 'data' InputSize: [224 224 3] Hyperparameters DataAugmentation: 'none' Normalization: 'zerocenter' NormalizationDimension: 'auto' Mean: [224×224×3 single]

inputSize=net.Layers（1）.inputSize；

最後の層の置き換え

ネットワークの畳み込み層は,入力イメージを分類するために,最後の学習可能な層と最終分類層が使用するイメージの特徴を抽出します。GoogLeNetのこれらの2つの層“loss3分类器”および“输出”は,ネットワークによって抽出された特徴を組み合わせてクラス確率,損失値,および予測ラベルにまとめる方法に関する情報を含んでいます。新しいイメージを分類するために事前学習済みのネットワークを再学習させるには,これら2つの層を新しいデータセットに適応させた新しい層に置き換えます。

学習済みのネットワークから層グラフを抽出します。ネットワークが AlexNet、VGG-16、VGG-19などの系列网络オブジェクトである場合,网。层の層のリストを層グラフに変換します。

如果isa（净额，“系列网络”)lgraph=层图（net.Layers）；其他的lgraph = layerGraph(净);结束

置き換える 2.つの層の名前を見つけます。これは手動で行うことも、サポート関数FindlayerStore酒店を使用してこれらの層を自動的に見つけることもできます。

[learnableLayer, classLayer] = findLayersToReplace (lgraph);[learnableLayer, classLayer]

ans=1×2带层的层阵列：1个“loss3分类器”完全连接1000个完全连接的层2个“输出”分类输出带“tench”和999个其他类的crossentropyex

ほとんどのネットワークでは、学習可能な重みを持つ最後の層は全結合層です。この全結合層を、新しいデータセットのクラスの数 (この例では 5) と同じ数の出力を持つ新しい全結合層に置き換えます。挤压网などの一部のネットワークでは、その代わりに最後の学習可能な層が 1 x 1畳み込み層となっています。この場合は、その畳み込み層を、クラスの数と同じ数のフィルターを持つ新しい畳み込み層に置き換えます。新しい層での学習速度を転移された層より速くするには、層の学習率係数を大きくします。

numClasses =元素个数(类别(imdsTrain.Labels));如果isa (learnableLayer'nnet.cnn.layer.FullyConnectedLayer'newLearnableLayer = fullconnectedlayer (numClasses，...“姓名”,“新俱乐部”,...“WeightLearnRateFactor”,10,...“BiasLearnRateFactor”10);埃尔塞夫isa (learnableLayer'nnet.cnn.layer.Convolution2DLayer')newLearnableLayer=卷积2dLayer（1，numClass，...“姓名”,“新康”,...“WeightLearnRateFactor”,10,...“BiasLearnRateFactor”10);结束lgraph=replaceLayer（lgraph，learnableLayer.Name，newLearnableLayer）；

分類層はネットワークの出力クラスを指定します。分類層をクラスラベルがない新しい分類層に置き換えます。列车网络は,学習時に層の出力クラスを自動的に設定します。

newClassLayer=classificationLayer(“姓名”,“新类输出”); lgraph=replaceLayer（lgraph，classLayer.Name，newClassLayer）；

新しい層が正しく結合されていることを確認するには,新しい層グラフをプロットして,ネットワークの最後の方の層を拡大します。

身材(“单位”,“正常化”,“位置”，[0.30.30.40.4]）；绘图（lgraph）ylim（[0,10]）

初期の層の凍結

これで、新しい一連のイメージでネットワークに再学習させる準備が整いました。オプションで、ネットワークの初期の層について学習率を 0に設定すると、それらの層の重みを "凍結" できます。学習中に列车网络は凍結された層のパラメーターを更新しません。凍結された層の勾配は計算する必要がないため、多数の初期の層について重みを凍結すると、ネットワーク学習を大幅に高速化できます。新しいデータセットが小さい場合、初期のネットワーク層を凍結すると、新しいデータセットに対するこれらの層の過適合を防止することもできます。

層グラフの層および結合を抽出し、凍結する層を選択します。水壶では、最初の 10個の層がネットワークの初期 'ステム' を作ります。サポート関数freezeWeightsを使用して、最初の 10個の層について学習率を 0に設定します。サポート関数createLgraphUsingConnectionsを使用して,すべての層を元の順序で再結合します。新しい層グラフに含まれる層は元と同じですが,初期の層の学習率は0に設定されています。

layers=lgraph.layers；connections=lgraph.connections；layers（1:10）=冻结权重（layers（1:10））；lgraph=createLgraphUsingConnections（layers，connections）；

ネットワークの学習

ネットワークにはサイズが 224x224x3の入力イメージが必要ですが、イメージ データストアにあるイメージのサイズは異なります。拡張イメージ データストアを使用して学習イメージのサイズを自動的に変更します。学習イメージに対して実行する追加の拡張演算として、学習イメージを縦軸に沿ってランダムに反転させる演算や、水平方向および垂直方向に最大 30ピクセルだけランダムに平行移動させて最大 10% スケールアップする演算を指定します。データ拡張は、ネットワークで過適合が発生したり、学習イメージの正確な詳細が記憶されたりすることを防止するのに役立ちます。

pixelRange=[-30]；ScalerRange=[0.9 1.1]；imageAugmenter=imageDataAugmenter(...“RandXReflection”是的，...“RandXTranslation”，像素范围，...“RandYTranslation”，像素范围，...“RandXScale”，鳞片，...“兰迪斯卡尔”, scaleRange);augimdsTrain = augmentedImageDatastore (inputSize (1:2), imdsTrain,...“数据增强”图像增强仪）；

他のデータ拡張を実行せずに検証イメージのサイズを自動的に変更するには,追加の前処理演算を指定せずに拡張イメージデータストアを使用します。

augimdsValidation=增强图像数据存储（inputSize（1:2），imdsValidation）；

学習オプションを指定します。InitialLearnRateを小さい値に設定して,まだ凍結されていない転移層での学習速度を下げます。上記の手順では,最後の学習可能な層の学習率係数を大きくして,新しい最後の層での学習時間を短縮しています。この学習率設定の組み合わせによって,新しい層では高速に学習が行われ,中間層では学習速度が低下し,凍結された初期の層では学習が行われません。

学習するエポック数を指定します。転移学習の実行時には、同じエポック数の学習を行う必要はありません。エポックとは、学習データセット全体の完全な学習サイクルのことです。ミニバッチのサイズと検証データを指定します。エポックごとに 1.回、検定精度を計算します。

miniBatchSize = 10;valFrequency =地板(元素个数(augimdsTrain.Files) / miniBatchSize);选择= trainingOptions (“个”,...“MiniBatchSize”miniBatchSize,...“MaxEpochs”6...“初始学习率”，3e-4，...“洗牌”,“every-epoch”,...“ValidationData”，augimdsValidation，...“ValidationFrequency”valFrequency,...“详细”假的,...“情节”,“培训进度”）;

学習データを使用してネットワークに学習させます。既定では、利用可能な GPUがある場合、列车网络はGPUを使用します(并行计算工具箱™,および以3.0计算能力上のCUDA®対応GPUが必要)。そうでない場合,列车网络は 中央处理器を使用します。trainingOptionsの名前と値のペアの引数“执行环境”を使用して、実行環境を指定することもできます。データセットのサイズが非常に小さいため、学習は短時間で終了します。

网= trainNetwork (augimdsTrain、lgraph选项);

検証イメージの分類

微調整したネットワークを使用して検証イメージを分類し、分類精度を計算します。

(YPred,聚合氯化铝)=(网络,augimdsValidation)进行分类;精度= mean(YPred == imdsvalidate . labels)

精度=0.9000

4枚のサンプル検証イメージを,予測ラベルとイメージがそれらのラベルを持つ予測確率と共に表示します。

idx = randperm(元素个数(imdsValidation.Files), 4);数字对于i=1:4子批次（2,2，i）i=readimage（imdsValidation，idx（i））；imshow（i）label=YPred（idx（i））；标题（字符串（标签）+”、“+num2str（最大100*3）+"%"）;结束