TensorFlow.js で RGB2GRAY を実装する

TensorFlow.js の記事自体が少ないので、どんどん生成していきたい気持ちです。

とても基本的な機能ですが、意外と公式ドキュメント見てもささっとは実装できなかったので記録。

// rgb は Tensor
const [r, g, b] = tf.split(rgb, [1, 1, 1], 2);
const gray = r.mul(tf.scalar(0.2126))
.add(g.mul(tf.scalar(0.7152)))
.add(b.mul(tf.scalar(0.0722)))

split、mul、add を使って js 風に繋げて行く感じですね。

Python の gray = 0.2126*rgb[0] + 0.7152*rgb[1] + 0.0722*rgb[2] みたいに、list[i] 形式が使えないのがなかなかもどかしい感じです。