基于轻量级深层卷积神经网络的花卉图像分类系统

摘要：

为解决深层卷积神经网络(Deep convolutional neural network,DCNN)模型在算力弱、存储成本高的AI边缘计算设备上难以高效应用的现实问题,本文利用重量级网络辅助训练轻量级网络,设计了一种基于轻量级神经网络的花卉图像分类系统.首先利用重量级DCNN并结合迁移学习、爬虫技术与最大连通区域分割方法,构建了适用于轻量级网络训练的扩充花卉数据集.然后基于Tiny-darknet与Darknet-reference两种网络及扩充后的花卉数据集训练得到两种面向弱算力设备的轻量级DCNN模型.训练得到的两种花卉分类网络在Oxford102花卉数据集上的平均分类准确率可达98.07％与98.83％,模型大小分别为4 MB与28 MB,在AI边缘计算设备中具有较好的应用前景.

关键词：花卉图像分类;深层卷积神经网络;深度学习

所属期刊栏目：36

分类号：TP391.41;TP183(计算技术、计算机技术)

资助基金：国家自然科学基金;湖北省中央引导地方科技发展专项基金

在线出版日期：2021-09-06（万方平台首次上网日期，不代表论文的发表时间）

页数：共13页

页码：756-768