This program is used to classify graphene
There are two folders in model_img, good and bad. Put pictures that has graphene in good folder, and the others put in bad folder, then execute graphenePredict_train.py to train model. The trained model will save in trained_model folder.
Put photos that you need to classfy in test_img, then execute graphenePredict_load.py to load the trained model. The program will print if the photo is good or bad.