计算机与现代化

• 图像处理 • 上一篇    下一篇

基于卷积神经网络的花朵图片分类算法

  

  1. (南昌航空大学江西省图像处理与模式识别重点实验室,江西南昌330063)
  • 收稿日期:2018-02-28 出版日期:2018-09-29 发布日期:2018-09-30
  • 作者简介:张小锋(1963-),男,江西南昌人,南昌航空大学江西省图像处理与模式识别重点实验室副教授,硕士生导师,博士,研究方向:计算机视觉,目标检测,图像处理; 刘红铮(1991-),男,河南开封人,硕士研究生,研究方向:图像处理,深度学习。
  • 基金资助:
    国家自然科学基金资助项目(61272077); 江西省自然科学基金资助项目(2014BA207012)

Flower Image Classification Based on Convolutional Neural Network

  1. (Key Laboratory of Jiangxi Province for Image Processing and Pattern Recognition, Nanchang Hangkong University, Nanchang 330063, China)
  • Received:2018-02-28 Online:2018-09-29 Published:2018-09-30

摘要: 针对目前利用卷积神经网络进行花朵图像分类时,全连接层产生的参数冗余和破坏空间结构信息问题,提出一种有效的改进方法。首先用1×n和n×1卷积核替换n×n卷积核,然后在卷积层后连接空间金字塔池化进行降维提取特征,最后在Softmax分类器输出概率分布。实验表明本文的方法不仅提高了准确率,而且使训练时间下降了一半,大大提高了训练的速度。

关键词: 卷积神经网络, 花朵图像分类, 全连接层, 空间金字塔池化

Abstract:  Aiming at the problem of parameter redundancy and destruction of spatial structure information produced by fully connected layer in flower image classification using convolution neural network, this paper proposes an effective improvement method. Firstly, the n×n convolutional filters are replaced by 1×n and n×1 convolutional filters, then they are connected to the spatial pyramid pooling behind convolution layer to reduce the dimension and extract features, finally the probabilistic distribution is exported in Softmax classifier. Experimental results show that this method not only improves the accuracy, but also reduces the training time by half, which greatly improves the training speed.

Key words: convolutional neural network, flower image classification, fully connected layer, spatial pyramid pooling

中图分类号: