Author: Caihao Cui Location: Melbourne, VIC, AU.
Description: This repository demonstrates one python program including training, testing, and retraining ML/DL models with TensorFlow and Keras with an extra GUI by Qt5.
Dependencies: install Anaconda python 3, TensorFlow, Keras and all the package related. TF2.0 works but with few warninig.
Use the Train_MNIST_Model.py to build the model, model will be saved at ./models/xxx.xxx.xxx.h5, the training histroy (acc, loss) will be saved as history.pickle.
Use the Test_MNIST_Model.py to test the model and compare the training and testing accuracy.
Use the Retrain_MNIST_Model.py to load the trained model and retrain it with different optimization methods.
Run GUI.py or MNIST_Window.py to play with the interactive desktop application.
You can change the details by editing the class of MNIST_Window
Sample Images:
This repository is inspried by the CSDN Blog: https://blog.csdn.net/u011389706/article/details/81460820.