2

セマンティック セグメンテーション用の FCN を作成しています。PascalVOC データセットで、ラベル付きの png 画像をインデックス付きの色の値に変換するのに苦労しています。値を 0 から 20 の範囲にしたい。次のコードで PIL を使用してこのような操作を実現できるため

with Image.open(image_path) as img:
    label = np.array(img)

それは私が望むものを出力します。しかし、テンソルフローの実装では、次のようなコードで同じ値にしたい

file = tf.read_file(image_path)
label = tf.image.decode_png(file, channels=0)

しかし、テンソルフローの実装では、0 から 255 までの値が返されます。テンソルフローでも PIL 実装を実現する方法はありますか? ありがとうございました。

4

1 に答える 1