代码链接
链接: https://github.com/JDHHH/Images
介绍
.ipynb文件是训练模型的,训练好了以后会生成一个包含模型结构和权重model.h5文件和一个只有模型权重的文件(可以根据自己喜好选择用哪个)。
.py文件是用tkinter简单写的UI,可以打开本地图片进行识别。两个.py文件都可以用,注意修改里面的模型文件地址。
在下跑了50轮的正确率达到了70%+,效果还可以。
跑模型是在Google colab跑的,kaggle上应该也可以。
(git clone的话,别忘了点个star!!!)