Handwritten Digit Recognition with CNN

Installation

Requirements

• Node.js ver. 10+.
• yarn

Clone this repository to your local computer

git clone https://github.com/dmnlgn/handwritten-digit-recognition-CNN.git

Configuration

The data.js file contains the absolute path to the dataset, so you have to change it in the cloned repository.

const datasetPath = "file:///GIT/portfolio/handwritten-digit-recognition-CNN/dataset/";

Training mode

Partial trainining

node app.js --train_mode=0 --epochs=10 --batch_size=64

Full training with saving the model

node app.js --train_mode=1 --epochs=10 --batch_size=64 --model_save_path=file://./public/assets/model

Front-end

Launch locally

yarn start

Open http://localhost:3000/ with your browser