カスタム学習ループを​用いたセグメンテーシ​ョンについて

カスタム学習ループを用いて0と1の値でセグメンテーションをしたいのですが、forward関数に通しても、0〜1のさまざまな数値が出てしまいます。forward関数に通した後はどのようなコードを記述するべきでしょうか？

Answer 1

[Tohru Kikawada]

下記のpixelClassificationLayerを使うことで出力を離散値にできると思います。 https://jp.mathworks.com/help/vision/ref/nnet.cnn.layer.pixelclassificationlayer.html

お試しいただければと思います。

[kaede]

遅れて申し訳ないですが、回答ありがとうございます。

以上の手法を試してみます。