Tensorflow.jsで手書き文字認識
以前から触ってみたかったTensorflow.jsを触ってみました。
今回は、モデル構築の部分は省いて、推論部分のみの実装です。
下記サイトを参考にさせていただきました。
TensorFlow.jsでMNIST学習済モデルを読み込みブラウザで手書き文字認識をする
Tensorflow.jsでmnist手書き数字認識をやってみた(React)
実装したコードは以下。
https://github.com/hisami/tensorflow-mnist-react
参考サイトと異なる点は主に以下2点です。
・クラスコンポーネントではなく関数コンポーネントで実装
・最新版のTensorflow.jsでは存在しない関数が使われていたりしたので(tf.fromPixelsなど)、最新版に合わせて修正
動作している様子は以下。なかなかの精度で推定できています。(構築済みのモデルが優秀なだけ。)
-
前の記事
DockerでJupyterLabを起動してグラフを表示する 2019.05.26
-
次の記事
Amazon Cognitoを使ってNode.jsで認証 2019.11.16