Tensorflow.jsで手書き文字認識

Tensorflow.jsで手書き文字認識

以前から触ってみたかったTensorflow.jsを触ってみました。
今回は、モデル構築の部分は省いて、推論部分のみの実装です。
下記サイトを参考にさせていただきました。
TensorFlow.jsでMNIST学習済モデルを読み込みブラウザで手書き文字認識をする
Tensorflow.jsでmnist手書き数字認識をやってみた(React)

実装したコードは以下。
https://github.com/hisami/tensorflow-mnist-react

参考サイトと異なる点は主に以下2点です。
・クラスコンポーネントではなく関数コンポーネントで実装
・最新版のTensorflow.jsでは存在しない関数が使われていたりしたので(tf.fromPixelsなど)、最新版に合わせて修正

動作している様子は以下。なかなかの精度で推定できています。(構築済みのモデルが優秀なだけ。)