Tensorflow.js digit recognition

A demonstration of using decapitated mobilenet for a digit recognition on the custom dataset.

Full image

We're interested in finding out top-right-most number.

Digit

Requirements

  • node
  • npm

Installation

npm i

Usage

Run:

npm run dev

Then open your browser and go to:

http://localhost:1234/train

After the model finishes the training, two json files should be automatically downloaded by your browser: pretrained-model.json and pretrained-model.weights.bin.

Train

Move these two files into the dist/model folder and then go to:

http://localhost:1234/test

After a bit of waiting you should be able to see the results of the training

Test

If you wanted to see the generator of the testing data, you can go to:

http://localhost:1234/generate

Generate

Built With

  • Tensorflow.js
  • React
  • Parcel

Authors