Using TensorFlow.js to train data and predict numbers just in your browser!
The training script is taken from tfjs-example.
screencast-dabuside_github_io-2024_07_09-18_33_13.webm
git clone repo
cp ./mnist_images.png ./dist/mnist_images.png & cp ./mnist_labels_uint8 ./dist/mnist_labels_uint8
yarn
yarn watch
yarn build
- tensorFlow.js requires webGL, you'd better use PC.
- you may wait 10s to load training set(~10mb) and training script(~1mb) and another 3s to train data.