长沙航空职业技术学院学报2024,Vol.24Issue(3) :27-31.DOI:10.13829/j.cnki.issn.1671-9654.2024.03.006

基于Android平台的柑橘检测算法研究

Research on Citrus Detection Algorithm Based on Android Platform

刘学 郑哲文 朱洪前
长沙航空职业技术学院学报2024,Vol.24Issue(3) :27-31.DOI:10.13829/j.cnki.issn.1671-9654.2024.03.006

基于Android平台的柑橘检测算法研究

Research on Citrus Detection Algorithm Based on Android Platform

刘学 1郑哲文 1朱洪前1
扫码查看

作者信息

  • 1. 中南林业科技大学材料科学与工程学院,湖南 长沙 410004
  • 折叠

摘要

针对采摘机器人在实际应用中所需要的视觉检测能力,提出一种基于深度学习的移动端柑橘检测算法,实现对柑橘类水果的快速实时检测.通过PyTorch框架将YOLOv5s、YOLOv5s6、YOLOv5m等模型在自制的柑橘数据集下进行训练,将训练好的模型转化为ONNX模型,再将ONNX模型转化为NCNN模型,并进行Android移动端部署.试验结果表明:NCNN深度学习框架下的YOLOv5s6识别率达到92.9%,召回率达到75.7%,能较好地满足采摘机器人实际应用需求.

Abstract

In view of the visual detection ability required by the picking robot in practical application,a mobile citrus detection algorithm based on deep learning was proposed to realize the rapid real-time detection of citrus fruits.Models such as YOLOv5s,YOLOv5s6,and YOLOv5m were trained on self-made citrus datasets through the PyTorch framework,and the trained models were transformed into ONNX models,and the ONNX models were converted into NCNN models,and deployed on Android mobile terminals.The experimental results show that the recognition rate of YOLOv5s6 under the NCNN deep learning framework reaches 92.9%and the recall rate reaches 75.7%,which can better meet the practical application needs of picking robots.

关键词

采摘机器人/深度学习/移动端/YOLOv5/NCNN

Key words

picking robot/deep learning/mobile terminals/YOLOv5/NCNN

引用本文复制引用

出版年

2024
长沙航空职业技术学院学报
长沙航空职业技术学院

长沙航空职业技术学院学报

影响因子:0.373
ISSN:1671-9654
参考文献量10
段落导航相关论文