首页|基于YOLOv4的水下海参检测与计数算法

基于YOLOv4的水下海参检测与计数算法

扫码查看
针对智慧水产养殖中海参自动采捕和高效计量应用需求,提出一种基于YOLOv4的水下海参检测与计数算法.该算法利用暗通道先验算法对数据集进行预处理,增强图像数据的可检测性;采用迁移学习方法训练YOLOv4网络,并用Swish函数替换骨干网络中的激活函数,提升自建数据集的海参检测性能;提出基于相近帧目标质心定位偏移的降重计数方法,优化目标计数结果.试验结果表明:该检测算法识别水下海参目标的平均检测精度的平均值mAP达91.0%,分别比原始YOLOv4、YOLOv3、Faster R-CNN和SDD高4.5%、6.9%、5.0%、29.9%;降重计数算法获得海参数量与人工计数结果间的均方根误差RMSE为29.8、平均计数精度ACP为95.8%、决定系数R2为0.998.
Recognition and counting algorithm of underwater sea cucumbers based on YOLOv4 network
In order to meet the requirements of automatic harvesting and high-efficiency measurement of sea cucumbers in intelligent aquaculture,a recognition and counting algorithm of underwater sea cucumbers based on YOLOv4 is proposed in this paper.The algorithm preprocesses the data set by using dark channel prior defogging algorithm to enhance the detectability of image data,YOLOv4 Network is trained with the transfer learning method,and Swish function is used to replace the activation function in the backbone network to improve the detection performance of the self-built data set,a method based on the target centroid positioning offsets of adjacent frames is proposed to optimize target counting result.The experimental results show that the mAP of sea cucumber targets recognized by the algorithm of this paper reaches 91.0%,which is 4.5%、6.9%、5.0%and 29.9%higher than that recognized by original YOLOv4,YOLOv3,Faster R-CNN,and SDD,respectively.The RMSE between the number of sea cucumbers obtained by reducing repeated counts and the manual counting result is 29.8.The average counting precision(ACP)is 95.8%and the coefficient of determination(R2)is 0.998.

sea cucumberdark channel priorYOLOv4transfer learningreducing repeated counts

宋小鹿、冯娟、梁翔宇、周玺兴

展开 >

河北农业大学信息科学与技术学院,河北保定,071001

河北省农业大数据重点实验室,河北保定,071001

河北农业大学机电工程学院,河北保定,071001

海参 暗通道先验 YOLOv4 迁移学习 降重计数

河北省重点研发计划项目

20327217D

2024

中国农机化学报
农业部南京农业机械化研究所

中国农机化学报

CSTPCD北大核心
影响因子:0.684
ISSN:2095-5553
年,卷(期):2024.45(9)
  • 4