基于卷积神经网络的油茶籽完整性识别方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2018YFDO700102-02)和赣南油茶产业开发协同创新中心开放基金项目(YP201611)


Integrity Recognition of Camellia oleifera Seeds Based on Convolutional Neural Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对现有油茶籽色选机无法识别碎籽的问题,提出一种基于卷积神经网络的油茶籽完整性识别算法。以油茶籽完整性识别为目标,构建油茶籽图像库;基于油茶籽完整性识别任务要求,通过对AlexNet网络进行优化得到适合油茶籽完整性识别的卷积神经网络模型,该网络具有4层卷积层、2层归一化层、3层池化层和1层全连接层。为了提高网络分类准确率和实时性,从网络结构简化和超参数优化两方面对卷积神经网络进行优化,最终网络结构(CO-Net)的分类准确率、训练收敛速度和泛化性能均得到了提高。实验结果表明,优化后的网络对油茶籽完整性识别准确率达98.05%,训练时间为0.58h,模型规模为1.65MB,单幅油茶籽图像检测平均耗时13.91ms,可以满足油茶籽在线实时分选的要求。

    Abstract:

    In order to solve the problem that the color sorter can not recognize the intact and the broken Camellia oleifera seeds, an integrity recognition algorithm of Camellia oleifera seed based on convolution neural network was proposed and the image database of Camellia oleifera seed was constructed. The network structure simplification and hyper-parameter optimization was conducted to improve the classification accuracy and real-time performance of the model. Firstly, the batch normalization (BN) layer of the model was selected by the comparison experiment, which speeded up the training of the model and improved the generalization performance of the model. Moreover, the Swish function was chosen as the model activation function, which improved the recognition accuracy and speeded up the convergence of the model. Furthermore, the depth and width of the network were changed to compress the size of the model and shorten the training time. In depth, the model included four convolution layers and one fully connected layer. And in the width, the number of local receptive fields (LRFs) in the convolution layers and the number of nodes in the fully connected layer were compressed. And the second and third convolution layers were replaced by the depthwise convolutions. After the structural improvement, the model was transferred to CO-Net, which was more suitable for the integrity identification of Camellia oleifera seeds. Besides, the hyper-parameters (batch size and learning rate) that affected the performance of the model were optimized. The final model (CO-Net) not only improved the classification accuracy but also speeded up the training convergence speed and enhanced the generalization performance of the model. The results showed that the accuracy of the optimized network was 98.05%, the training time was only 0.58h, and the model size was only 1.65MB. The average time of detecting an image of Camellia oleifera seed was only 13.91ms, which can meet the requirements of realtime sorting of Camellia oleifera seed.

    参考文献
    相似文献
    引证文献
引用本文

谢为俊,丁冶春,王凤贺,魏硕,杨德勇.基于卷积神经网络的油茶籽完整性识别方法[J].农业机械学报,2020,51(7):13-21. XIE Weijun, DING Yechun, WANG Fenghe, WEI Shuo, YANG Deyong. Integrity Recognition of Camellia oleifera Seeds Based on Convolutional Neural Network[J]. Transactions of the Chinese Society for Agricultural Machinery,2020,51(7):13-21.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-04-16
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2020-07-10
  • 出版日期: