基于双模型决策级融合的鱼道分布外目标检测方法

Out-of-distribution fish detection in fishways based on decision-level fusion of two models

  • 摘要: 针对鱼道过鱼目标检测鱼类特征模糊及数据集先验信息不足而导致的漏检,误检问题,研究了1种基于粗-细粒度模型决策级融合的目标检测方法。该方法首先对粗粒度YOLO模型进行改进:在主干层嵌入坐标注意力模块,在特征融合部分嵌入自适应特征融合模块对不同尺度的特征层进行融合,以提高粗粒度模型对任意鱼类的检测能力;最后将改进后的粗粒度YOLO模型与细粒度YOLO模型的检测结果按照置信度筛选出需要融合的检测框,并按照置信度的值对检测框进行加权融合,以此降低在未知鱼类、模糊鱼类场景下进行分布外检测时的漏检率和误检率。应用此方法于真实环境下的鱼道过鱼数据集进行测试,对未知鱼类的准确率达到了98.59%,召回率达到了94.19%,相比基于置信度的分布外检测方法分别提高了9.25%和11.21%,相比基于能量的分布外检测方法分别提高了6.42和3.69%。对模糊目标的识别准确率达到了95.45%,召回率达到了91.8%,相比基于置信度的分布外检测方法分别提高了16.63%和18.58%,相比基于能量的分布外检测方法分别提高了11.27%和1.74%。研究成果对鱼道过鱼的目标检测有良好的借鉴意义。

     

    Abstract: To address the issues of missed detection and false alarms resulting from ambiguous fish features and insufficient prior information in the dataset for fish detection in fishways, this paper proposes an object detection algorithm founded on decision fusion between coarse-grained and fine-grained YOLO models. The method first enhances the coarse-grained YOLO model by embedding a coordinate attention module into the backbone layers and an adaptive spatial feature fusion module in the feature aggregation part to fuse features from different scale layers, thereby improving the model's detection capability for arbitrary fish species. Finally, the detection outcomes from the refined coarse-grained YOLO model and fine-grained YOLO model are filtered by confidence scores to identify boxes that need fusion. These boxes are then weighted fused based on their confidence values. This allows the proposed approach to attain a lower miss rate and false alarm rate for out-of-distribution detection on unknown and blurry fish. Evaluated on real-world fishway fish datasets, The proposed method achieves 98.59% accuracy and 94.19% recall for unknown fish, which are 9.25% and 11.21% higher than the confidence-based out-of-distribution detection method, and 6.42% and 3.69% higher than the energy-based out-of-distribution detection method, respectively. The recognition accuracy of fuzzy targets reaches 95.45% and the recall rate reaches 91.8%, which is 16.63% and 18.58% higher than the confidence-based out-of-distribution detection method, and 11.27% and 1.74% higher than the energy-based out-of-distribution detection method, respectively. The research findings have valuable implications for fish detection in fishways.

     

/

返回文章
返回