关键词:
非法采砂
图像去雾
生成对抗网络
目标检测
YOLOv8算法。
摘要:
为解决长江流域非法采砂船只试别困难、监测效率低、精度差等问题,提出一种基于深度学习的雾天非法采砂船只辨识方法。首先,提出改进的生成对抗网络对采集图像进行去雾处理,得到清晰水域图像。其中,利用融合特征注意力机制的生成器提取雾天环境下船只的复杂纹理特征;在判别器中加入谱归一化操作,解决网络训练梯度消失的问题;改进损失函数,引入循环一致性损失保证生成图像与原始图像间的结构一致性。其次,提出融合注意力机制的YOLOv8算法,有效增强网络对图像重要特征的提取能力,实现对清晰水域图像中非法采砂船只的精准定位与辨识。实验结果表明,本研究提出的改进生成对抗网络对图像的去雾效果较好,峰值信噪比(PSNR)与结构相似性(SSIM)分别为31.86和0.64,较Cycle GAN和GC-GAN算法分别提升了3.6%~13.1%、4.9%~56.1%。去雾后的图像经过融合注意力机制的YOLOv8算法处理,可实现对非法采砂船只的准确识别与准确定位,其mAP@0.5:0.95和帧率可达到89.6%、36 FPS,满足公安实战对精度与速度的要求,可有效提高长江流域非法采砂信息化、智能化监管与执法水平。