针对目前难以在复杂场景下实现鱼群计数的问题，提出一种基于改进YOLOv5的鱼群计数方法.该算法是在YOLOv5s的基础上，采用k-means++优化聚类先验框替代原算法中的先验框，以提高对小目标鱼群的识别能力，并引入Soft-NMS策略选取最终预测框，对鱼群目标进行更加有效的检测.实验结果表明，经自制的鱼群检测数据集验证，改进后的算法平均精度均值可达96.29%.
科研之友
微信
新浪微博
Facebook
分享链接
