主管单位:吉林省科技厅
主办单位:吉林省科学技术信息研究所
合作单位:吉林省科学技术情报学会

农业与技术 ›› 2024, Vol. 44 ›› Issue (2): 23-27.DOI: 10.19754/j.nyyjs.20240130007

• 农业工程 • 上一篇    下一篇

基于迁移学习和数据增强的小样本柑橘缺陷检测研究

贾惠 伍希志 薛洋 李玉彤   

  1. 中南林业科技大学材料科学与工程学院,湖南长沙410000
  • 出版日期:2024-01-30 发布日期:2024-01-30
  • 作者简介:贾惠(1996-),女,硕士在读。研究方向:机器视觉,深度学习;通讯作者伍希志(1985-),博士,副教授。研究方向:人工智能及机器视觉,智能装备。
  • 基金资助:
    湖南省科技特派员服务乡村振兴“基于机器视觉与深度学习的重组竹板颜色分类技术研究”(项目编号:2023NK4285);中国博士后科学基金资助(项目编号:2021M690768)
  • Online:2024-01-30 Published:2024-01-30

摘要: 小样本目标检测旨在通过少量样本实现对图像中目标的识别和定位。目前针对柑橘缺陷的小样本数据集 进行的检测较少,本文提出了使用数据增强和迁移学习来对小样本柑橘的缺陷类型进行检测的方法,采用旋转、裁剪和高斯模糊进行数据增强来扩充数据集,与使用迁移学习方法进行对比。实验表明:迁移学习方法中最优算法是FRCN ft-fl山,基于该算法的20-shot任务的mAP值为67.823%;在基于数据增强的方法中使用Faster R-CNN算法的mAP值达到了84.7%,使用YOLOv8算法的mAP是85.3%,YOLOv8算法略优于Faster R-CNN算法。迁移学习方法增强了检测模型的泛化能力,加快了模型的收敛速度;数据增强方法有效扩充了数据集,提升了小样本柑橘缺陷检测模型的准确性。

关键词: 小样本识别;迁移学习;缺陷检测;数据增强;深度学习

中图分类号: