セマンティックセグメ​ンテーションの出力結​果をピクセル毎の確率​で表示する方法

FCNを用いたネットワークを作成し、セマンティックセグメンテーションの学習を行いました。

学習は概ねうまくいき、テスト用画像でセマンティックセグメンテーション を行わせてみたところ、添付した画像のように出力を得ることができました。しかしながらこの画像からは各ピクセルがどのクラスに分類されたかという情報しか得られません。セマンティックセグメンテーション の出力結果はネットワークの構造上、「ピクセルがあるクラスに属する確率」になると思うのですが、これを表示する方法はあるのでしょうか？

またその確率が一定の値以上であればこのクラスだと判定するという、いわゆる閾値のようなものが存在していると思うのですがそれを変化させて、結果を画像として出力することは可能でしょうか？

Answer 1

[Kenta]

こんにちは、activations関数を使えばいけます。

https://jp.mathworks.com/matlabcentral/fileexchange/75356-crack-detection-using-one-class-svm-1-svm

例えば、この例だと、activations(net,augimdsTest,'pool5'...(略))のように、やってますが、こちらの例だと、softmaxの後までactivationsすると、おそらく各クラスの確率に相当するものが吐き出せて、.7以上なら「ｘｘとする」のように条件分けすればできると思います。