deep-learning - 確率マップから確実なセグメンテーションを取得するにはどうすればよいですか?

Question

自分のデータ用にネットワークをトレーニングしました。これを実行したら：確率マップを表示するには

plt.imshow(prob[1], cmap='hot', interpolation='nearest')
plt.show()

そして、これは の確率マップですObject 1:

確率マップから確実なセグメンテーションを取得するにはどうすればよいですか (クラスに属する確率が低いポイントを考慮しても)?

ありがとう