直接看效果图:
cifar100数据集进行训练,得到的模型在可视化的界面上识别。
代码仓库和演示视频地址:
097cifar100基于卷积神经网络的识别_哔哩哔哩_bilibili
数据集包含100个类别,每个类别各500张图片,合集5w张和上一期的cifar10数据集图片个数一致。
同样每个类别也都是32*32尺寸的彩图
运行python 01数据集文本生成制作.py 可以生成txt文本保存在logs文件下,里面存在图片路径和对应标签。
运行python 02train.py会对data文件里面图片进行提取训练模型,得到的模型也保存在logs文件里。
运行03pyqt.py可以展示可视化的界面,加载图片调用模型识别结果。
感兴趣的小伙伴可以进行下载。