摘要
我国每年有大量人员因为非法在水库及河湖水域游泳、钓鱼溺水而丧生,这些水域往往地处偏僻,无法安排监管人员24小时值守。同时,由于速度和准确度难以兼具,或体积较大难于部署等原因,现有基于视觉的目标检测方法也无法实现对非法野泳、野钓行为的实时检测。基于此,提出了融合注意力机制的YoloX-ECA模型,通过在YoloX骨干网络中的残差块和特征金字塔网络中添加ECA模块,以求在保持原YoloX模型较快检测速度的同时提升对野泳、野钓行为的检测效果。基于自制野泳、野钓数据集的实验证明,改进的YoloX-ECA模型对河湖水域的野泳、野钓行为的检测性能(AP)在90%以上,检测速度为62.29fps。模型整体性能(mAP)较原YoloX模型提高1.21%,同其他目标检测算法如Faster-RCNN相比性能同样占优。改进的YoloX-ECA模型的实时性和准确性均达到预期设计目标,在河湖流域智能监管等领域有较大的应用前景。
Every year in China,a large number of people die from drowning due to illegal swimming and fishing in reservoirs,rivers,and lakes.These bodies of water are often located in remote areas,making it difficult for staff to supervise them 24 hours a day.Existing target detection methods are either too slow or inaccurate,too large to deploy,or incapable of detecting illegal swimming and fishing in real-time.To address these shortcomings,we proposed the YoloX-ECA model with an attention model.By adding the efficient channel attention(ECA)block to the CSPLayer and FPN,we aimed to improve detection performance for swimming and fishing while maintaining detection speed.Experimental results on self-made datasets showed that the YoloX-ECA achieved over 90%AP for the detection of swimming and fishing classes,with a detection speed of 62.29 fps.Compared with YoloX,mAP was increased by 1.21%.Furthermore,YoloX-ECA’s performance also outperformed other target detection algorithms such as Faster-RCNN.The improved YoloX-ECA model achieved the expected design goals and displayed great prospects for application in the field of intelligent supervision of rivers and lakes.
作者
罗文宇
傅明月
LUO Wen-yu;FU Ming-yue(School of Electronic Engineering,North China University of Water Resources and Electric Power,Zhengzhou,Henan 450046,China;School of Information Engineering,North China University of Water Resources and Electric Power,Zhengzhou,Henan 450046,China)
出处
《图学学报》
CSCD
北大核心
2023年第3期465-472,共8页
Journal of Graphics