博客:https://blog.csdn.net/qq_34904125/article/details/124854899
本篇是python语言写的,pytorch框架,基于mobilenet训练水果数据集,数据集也可以自行替换,
文件夹下放对应的图片即可,通过01demo即可生成txt文本的数据集,存放图片路径和标签。
02demo使用迁移学习的mobilenet卷积网络对图像进行训练模型,
03demo是通过flask接口调用模型,与小程序界面参数交互
代码中有说明文档
然后小程序的完整上传图片代码也在里面,都是可以直接使用的。