Cake_Mooncake

main_page_0

基于tensorflow2.5的糕点分辨系统

本项目在fork至宋老狗 在gitee上的相关代码。我是一名大一AI专业的学生,在暑假闲来无事想跑一跑tenorflow,正是这位老哥的相关教程让我一步一步入坑深度学习,PS:之后相关知识会认真学的,暑假只是想体验下tensorflow这个框架,感受下训练模型是什么感觉。

我们班在今年6月刚刚分班,我是本班班长,在最后一次聚会上我们用班费一起订了一个大蛋糕,上边写满了我们的名字,班主任的名字在**被围聚着,在吃完蛋糕后我想能不能自己训练一个OCR文字识别系统让它可以认出蛋糕上的名字,但后来觉得对于初学者有点困难于是索性就训练个可以认出这是个蛋糕就行。

识别结果

使用方法

可以利用文件中的get_data.py来爬取相应的图片,之后利用data_spit.py对图片进行分类,用tran_cnn,py或train_mobilenet.py进行进行训练,最后windows.py可以直接生成对应的判断界面。