/digit-classification

Primary LanguagePython

DIGIT CLASSIFICATION

Setup

For NVidia GPU

-Nvidia Drivers 410.x or higher

-CUDA 10.0

-CuDNN 7.4.1

-TensorRT 5.0

-pip packages pip3 install -r gpu.txt

For others

-pip packages pip3 install -r nogpu.txt

Import images

./importImage.sh <jpg/png file to import with path> <filename with extension> <label>