详细信息
Weighted multi-error information entropy based you only look once network for underwater object detection ( SCI-EXPANDED收录 EI收录) 被引量:2
文献类型:期刊文献
英文题名:Weighted multi-error information entropy based you only look once network for underwater object detection
作者:Ma, Haiping[1];Zhang, Yajing[1];Sun, Shengyi[1];Zhang, Weijia[1];Fei, Minrui[2];Zhou, Huiyu[3]
机构:[1]Shaoxing Univ, Dept Elect Engn, Shaoxing 312000, Zhejiang, Peoples R China;[2]Shanghai Univ, Sch Mechatron Engn & Automat, Shanghai Key Lab Power Stn Automat Technol, Shanghai 200072, Shanghai, Peoples R China;[3]Univ Leicester, Sch Comp & Math Sci, Leicester LE1 7RH, England
年份:2024
卷号:130
外文期刊名:ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE
收录:SCI-EXPANDED(收录号:WOS:001149720200001)、、EI(收录号:20240115310567)、Scopus(收录号:2-s2.0-85181134289)、WOS
基金:This article was supported in part by the National Natural Science Foundation of China under Grant No. 61640316, and the Zhejiang Provincial Natural Science Foundation of China under Grant No. LY19F030011.
语种:英文
外文关键词:Underwater object detection; Information entropy; YOLO network; Illumination noise; Multi-error
外文摘要:Underwater object detection is considered as one of the most challenging issues in computer vision. In this paper, a weighted multi-error information entropy based YOLO (You Only Look Once) network is proposed to address underwater illumination noise affecting the detection accuracy. First, underwater illumination is essentially structural and non-uniform, and it is modeled as an independent and piecewise identical distribution, which is a generic noise model to describe the complex underwater illuminating environment and accommodates the traditional Gaussian distribution as a special case. Second, assisted by the proposed illumination noise model, a minimum weighted error entropy criterion, which is an information-theoretic learning method, is introduced into the loss function of YOLO network, and then the network parameters are trained and optimized to improve the detection performance. Furthermore, a multi-error processing strategy is simultaneously used to handle vector errors during information back-propagation in order to accelerate convergence. Experiments on underwater object detection datasets including URPC2018, URPC2019 and Enhanced dataset, show the proposed weighted multi-error information entropy based YOLOv8 network gets mean average precision (MAP) of 88.7%, 91.8% and 96.7% respectively, and average frames per second (FPS) of 116.6. These two evaluation metrics are better than the baseline YOLOv8 and the existing advanced non-YOLO approaches by at least 5.2% and 5.3% respectively. The results verify the effectiveness and superiority of the proposed network for underwater object detection in complex underwater environment.
参考文献:
正在载入数据...