SegNetを用いた​セマンティックセグメ​ンテーションの転移学​習

9 views (last 30 days)
Hirotada Masuda
Hirotada Masuda on 4 Jun 2018
Answered: Kei Otsuka on 5 Jun 2018
こちらのチュートリアルに基づき、Segnetを用いたセマンティックセグメンテーションを行っています。 https://jp.mathworks.com/help/vision/examples/semantic-segmentation-using-deep-learning.html
1. 最終的に4クラスでのセグメンテーションを行いたいのですが、そのうち2つずつ類似性を有するため、最初に4クラスを2クラスに集約してトレーニングし、その後4クラスに分類するという2段階トレーニングを行いたいと思っています。 SegNetはネットワークの生成時に分類するクラス数を指定することになっていますが、ネットワークの1回目の学習後に分類するクラス数を変更するにはどのようにすればよいでしょうか。 もしくは1段階目にトレーニングした2クラスSegnetモデルを、新しく作成した4クラスSegnetに取り込み転移学習することは可能でしょうか。
2. Segnetのトレーニング中に、トレーニングデータとはサイズの違うイメージおよびラベルデータを検証に用いる方法はありますでしょうか。
  1 Comment
michio
michio on 4 Jun 2018
URLをハイパーリンク化いたしました。

Sign in to comment.

Accepted Answer

Kei Otsuka
Kei Otsuka on 5 Jun 2018
#1. 2段階トレーニングについては経験がありませんが、FCN構築においてAlexnetからWeight/Biasを転移させた例が こちらにありますので、参考にしてみてください。
クラス数を変更する場合、クラス数に関わるレイヤは定義し直す必要があります。
 
#2. 検証用のデータはimageDatastoreで与えることが出来るようになっていますが、'ReadFcn'を使うことで、画像読み出し時に利用する関数を定義できます。imresizeなどの関数を利用して、ネットワークのInputSizeにリサイズしてお使いください。

More Answers (0)

Community Treasure Hunt

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

Start Hunting!