湖北农业科学 ›› 2024, Vol. 63 ›› Issue (8): 28-34.doi: 10.14088/j.cnki.issn0439-8114.2024.08.006

• 图像图形识别 • 上一篇    下一篇

基于深度可分离卷积的果蔬分类识别方法

岳振1, 李卓然1, 王绪谦1, 侯宗升1, 苗壮2, 郑毅3, 刘杰1   

  1. 1.青岛农业大学理学与信息科学学院,山东 青岛 266109;
    2.青岛鼎信通讯股份有限公司,山东 青岛 266109;
    3.平邑县毅文家庭农场,山东 临沂 273302
  • 收稿日期:2023-05-08 出版日期:2024-08-25 发布日期:2024-09-05
  • 通讯作者: 刘 杰,男,山东临沂人,副教授,主要从事智慧农业研究,(电话)18765282829(电子信箱)1831440117@qq.com。
  • 作者简介:岳 振(1988-),男,山东临沂人,讲师,博士,主要从事基于深度学习的农产品智能识别的研究工作,(电话)18766239659(电子信箱)1511052807@qq.com。
  • 基金资助:
    山东省自然科学基金面上项目“基于图自注意网络的图嵌入聚类技术研究”(ZR2021MF078); 青岛农业大学博士课题“农产品分选中的轻量级深度学习模型研究”

Fruit and vegetable classification and recognition method based on Depthwise Separable Convolution

YUE Zhen1, LI Zhuo-ran1, WANG Xu-qian1, HOU Zong-sheng1, MIAO Zhuang2, ZHENG Yi3, LIU Jie1   

  1. 1. School of Science and Information Science, Qingdao Agricultural University, Qingdao 266109, Shandong, China;
    2. Qingdao Topscomm Communication Co., Ltd., Qingdao 266109, Shandong, China;
    3. Yiwen Family Farm in Pingyi County, Linyi 273302, Shandong, China
  • Received:2023-05-08 Published:2024-08-25 Online:2024-09-05

摘要: 针对农贸市场、果蔬超市中结算流程不够智能化以及重型神经网络模型部署困难等问题,对果蔬分类模型轻量化识别方法进行了研究。首先针对果蔬智能识别设备所在环境差异大、果蔬套袋问题,采用多场景采集方案在果蔬超市现场采集果蔬170种、图片136 000张,并设计了弱化套袋的图像预处理方法,对数据进一步增强。然后针对重量级神经网络部署困难以及成本较高的问题,设计了一种基于深度可分离卷积的果蔬分类识别模型,并进行训练测试,其Top-1准确率达96.8%,Top-5准确率达100%,相对于Mobilenetv2-224,运算量减少了70%,相对于Mobilenetv3-224,运算量减少了60%,识别能力介于Mobilenetv2-224和Mobilenetv3-224之间。最后对所设计果蔬分类模型在实际部署中面临的问题进行了分析。

关键词: 果蔬分类, 图像增强, 深度可分类卷积, 轻量化神经网络

Abstract: Aiming at the problem that the settlement process in agricultural trade markets and fruit and vegetable supermarkets was not intelligent enough and the difficulty of deploying heavy neural network models, the lightweight recognition method of fruit and vegetable classification model was studied. Firstly, in response to the large differences in the environment where the intelligent recognition equipment for fruits and vegetables was located, and the problem of fuzzy features in fruit and vegetable bagging, a multi-scene collection scheme was used to collect 170 kinds of fruits and vegetables and 136 000 pictures in the fruit and vegetable supermarket, and an image preprocessing scheme for weakened bagging was formulated to further enhance the data. Secondly, aiming at the difficulty of deploying the heavyweight neural network and the high cost, a fruit and vegetable classification recognition model based on Depthwise separable convolution was designed, trained and tested. Its Top-1 success rate had reached 96.8%, and the Top-5 success rate had reached 100%. Compared to Mobilenetv2-224, the amount of computation had been reduced by 70%, compared to Mobilenetv3-224, the amount of computing had also been reduced by 60%, and the recognition ability was higher than Mobilenetv2-224 and lower than Mobilenetv3-224. Finally, the problems faced by the designed fruit and vegetable classification model in the actual deployment were analyzed.

Key words: fruit and vegetable classification, image enhancement, depthwise separable convolution, lightweight neural network

中图分类号: