卷积网路CNN分类的模型一般使用包括alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet和Swin transformer等10多种模型
本次是中草药的小程序识别
代码演示可以看bilibili009期
https://space.bilibili.com/124080712?spmidfrom=333.999.0.0
效果图如下:
代码文件包括算法部分和小程序部分
算法部分包括中草药的数据集文件夹下放的是不同类别的中草药文件夹,搜集对应的中草药图即可
01训练数据集文本生成.py会对中草药图像的路径进行保存到txt文本中
再使用02resnet迁移学习.py读取txt文本中的数据进行训练,得到的模型保存在当前文件夹下为model.ckpt文件
最后通过03flask_server.py开启服务与小程序端进行交互
然后是小程序部分
小程序部分就是设置了界面然后提供了上传图像或拍照的接口将获取的图像传给服务端识别结果再接收
欢迎下载