卷积网路CNN分类的模型一般使用包括alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet和Swin transformer等10多种模型
视频教程找036-037期:
https://space.bilibili.com/124080712?spmidfrom=333.1007.0.0
完整的文件展示如下:
算法部分
其中中药饮片数据集文件夹下存放的图像数据集。
运行01训练数据集文本生成.py会将数据集图片路径带上标签保存在txt文本中。
运行02train.py会将txt文本中的图像数据读取进行模型的训练,最后保存在weights文件夹下。
运行03flask_server.py是一个连接小程序端的接口,通过小程序交互图片识别。
欢迎下载