代码下载和视频演示地址:
079基于深度学习的香蕉成熟度检测小程序版_含10多种模型_哔哩哔哩_bilibili
效果展示图如下:
代码文件展示如下:
运行01数据集文本生成制作.py可以读取图片路径保存再txt文本中,
运行02train.py可以对txt文本中的图片路径读取并训练模型,
在02中可以选择的模型有10多种可以对比,包括alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet和Swin transformer等10多种模型。
训练的每个epoch都会显示准确率precision、召回率recall和 f1-score