基于YOLOV3模型的甘蔗丛环境下行人检测方法 |
| |
引用本文: | 邓敏,黄世醒,黄燕娟,郑丁科,张祺睿,杨丹彤.基于YOLOV3模型的甘蔗丛环境下行人检测方法[J].农机化研究,2023(1):8-14+57. |
| |
作者姓名: | 邓敏 黄世醒 黄燕娟 郑丁科 张祺睿 杨丹彤 |
| |
作者单位: | 华南农业大学南方农业机械与装备关键技术省部共建教育部重点实验室 |
| |
基金项目: | 国家现代农业产业技术体系(糖料)建设专项(CARS-17); |
| |
摘 要: | 针对在“一垄双沟”甘蔗丛高秆作物环境下行人目标检测的重要性及难度问题,提出一种改进YOLOV3算法的甘蔗丛中高秆作物遮挡的行人目标检测算法(YOLOV3-my-prune)。为获得更好的行人特征表达,使用Imgaug库对自制的数据集增强,并结合数据集中行人尺寸特点,使用K-means聚类分析方法,为神经网络重新聚类目标锚箱。对YOLOV3网络改进,设计了在网络全连接层中引入空间金字塔池化模块,并增加第四尺度预测特征图,同时增强网络多尺度特征融合能力及小尺度特征提取能力。完成改进的模型基础训练后,采用通道和层剪枝混合剪枝方法轻量化模型,并在数据集上进行多尺度训练并测试,结果表明:此方法 准确率达80.1%,检测速度为30fps,均有所提升。
|
关 键 词: | 行人检测 YOLOV3 空间金字塔池化网络 第四尺度特征图 甘蔗 剪枝 |
|
|