learning gluon with mnist dataset
python3 train.py
Need to modify the model and picture path in the code
python3 predict.py
the code Calculate 1000 times to average, if you only need to calculate once, comment it out in the code.
- mxnet
- mxboard
- time
- cv2
- numpy