003使用深度学习方法检测水果小程序版本

效果图如下：

​

代码演示和demo仓库看b站视频：

到此一游7758258的个人空间哔哩哔哩bilibili

代码展示：

​

数据集图片放置在data文件夹下，大家可以根据自己需要比如识别其他物体，只需要模仿data文件夹下的文件命名放入图片即可运行训练模型了。

​

 ​

 运行01数据集文本生成制作.py可以将data文件夹下的图片保存成txt格式

​

 运行02train.py可以将txt中记录的数据读取并训练模型

​

训练可以使用的卷积神经网络包括： efficientnet、Alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNext、ShuffleNet、Swin_transformer、VGG等。

​

 训练结束后同时会生成评价指标图

​

运行03flask_server.py可以生成一个http接口连接小程序端和代码端，将小程序传来的图片调用本地训练好的模型识别之后再返回给小程序界面展示。