公开(公告)号：CN115439744A

公开(公告)日：2022-12-06

申请号：CN202211038343.8

申请日：2022-08-29

Applicant: 华南农业大学

Inventor： 吴伟斌 ,  李杰 ,  赵新 ,  姚焙火 ,  高昌伦 ,  孙顺利 ,  韩重阳 ,  唐婷 ,  吴贤楠 ,  沈梓颖 ,  刘逸飞

IPC: G06V20/10 ,  G06V10/82 ,  G06V10/762 ,  G06N3/08 ,  G06N3/04

Abstract: 本发明公开了一种基于目标检测的复杂环境下茶叶嫩芽轻量化检测方法，包括：S1、收集茶叶嫩芽的图像，构建原始数据集；S2、使用图像标注工具对原始数据集中的茶叶嫩芽的一芽一叶和一芽两叶进行人工标注；S3、对YOLOv4进行轻量化改进，采用GhostNet作为主干特征提取网络，在YOLOv4的PANet中添加通道注意力机制；S4、对茶叶嫩芽图像数据集进行聚类分析获取符合茶叶数据集的anchor；S5、对改进后的YOLOv4设定参数，进行训练并保存最优模型；S6、将待检测的图像输入最优模型中进行前向推理，再经非极大值抑制后处理得出待检测的图像中茶叶嫩芽。本发明改进目标检测模型YOLOv4的主干特征提取网络部分，添加通道注意力机制，实现对复杂茶园环境下茶叶嫩芽的轻量化准确检测。