mnist-react-tfjs
v0.1.2
Published
This repo is a POC about using a trained Keras model with MNIST database with TensorFlow.js and React.js!
Downloads
6
Readme
mnist-react-tfjs
Number recognition with MNIST data (React.js and TensorFlow.js)
Predict numbers using MNIST database with TensorFlow.js and React.js!
author: Aral Roca [email protected]
DEMO
- http://aralroca.github.io/MNIST_React_TensorFlowJS
Requirements
- react - Recommended ^16.8.6
- react-dom - Recommended 16.8.6
- @tensorflow/tfjs - Recommended ^1.1.0"
Getting started
npm install --save mnist-react-tfjs
Then:
import NumberRecognition from 'mnist-react-tfjs';
Code
With RenderProps:
<NumberRecognition
width={400}
height={400}
>
{
(predictedNumber, predictions) => (
<h1>{predictedNumber}</h1>
)
}
</NumberRecognition>
Without RenderProps:
<NumberRecognition
width={400}
height={400}
onPredict={
(number, predictions) =>
console.log(number, predictions)
}
onClear={() => console.log('onClear')}
/>