Tensorflowでテンソルをプロットして保存する方法

Question

ネットワークの出力が次元（375,1242,1）のテンソルであるPythonを使用して、Tensorflowの完全畳み込みネットワークを推論しています。これは、特定のクラスに属するすべてのピクセルの確率を保持する出力イメージです（私の例ではRoadクラス - KITTY）。テンソルの形式はTensor("Slice:0", shape=(375, 1242, 1), dtype=float32)です。私の質問は、このテンソルを画像としてプロットして保存するにはどうすればよいのですか？このような何かをバイナリに変換するにはどうすればいいですか？thres=0.5, image = image > thres？

Answer 1

この質問はすでに複数回回答されています（たとえばmy answerを参照）。

オープンセッションを使用してnumpy配列を取得するには、まずテンソルを評価する必要があります。一度それを得たら、np_array=np_array[:,:,0]のようなことをして余分な次元を取り除かなければなりません。

次に、matplotlibを使用して、デフォルトでimshow(np_array)を実行すると、デフォルトでそれがカラーマップになり、正規化されます。

バイナリが必要な場合は、binary_array=(np_array>0.5).astype("int")とすれば、最終的にimshow(binary_array)とすることができます。