基于UNet的轻量SAR图像溢油检测方法
投稿时间:2024-03-18  修订日期:2024-05-30  点此下载全文
引用本文:
摘要点击次数: 68
全文下载次数: 0
作者单位邮编
江蒋伟* 江苏科技大学 214444
魏雪云 江苏科技大学 
唐志勇 江苏科技大学 
陈思远 中国电信股份有限公司盐城分公司 
基金项目:国家自然科学基金项目(面上项目,重点项目,重大项目)
中文摘要:针对传统深度学习模型因其参数量过大导致难以适应海洋溢油监测的边缘计算任务和溢油区域形态差异较大、边界难以分割等问题,尝试提出一种基于编码器-解码器结构和组合注意力机制的轻量级神经网络对溢油区域进行语义分割。首先在UNet网络的基础上对模型进行轻量化改良,降低模型参数量和计算复杂度。同时引入组合注意力机制,增强对溢油区的权重,抑制SAR图像固有的斑点噪声。此外,为了能更好地将溢油边界分割,本文还利用边界度量对损失函数进行改良。实验结果表明,该方法的像素准确率达到90.68%,参数数量仅为1.52M,图像推理速度为28.53毫秒。与UNet、Attention UNet、FCNet、PSPNet、DeeplabV3 Net等溢油图像分割模型进行比较,在保持分割质量相近的条件下,该模型参数量和推理速度都达到了较优水平。
中文关键词:溢油检测  轻量化  注意力机制  损失函数
 
Lightweight SAR image oil spill detection method based on UNet
Abstract:Incorporating advanced semantic segmentation techniques into edge computing tasks for monitoring marine oil spills is a formidable challenge due to the oversized model parameters of traditional deep learning models and the intricate segmentation requirements of oil spill boundaries. To mitigate this issue, this paper proposed a lightweight neural network based on encoder-decoder structures with a combined attention mechanism that can accurately segment the semantics of the oil spill area. Firstly, this paper took advantage of UNet's architecture and made subtle optimizations to reduce model parameters while maintaining computational efficiency. Additionally, a combined attention mechanism introduced in the paper that can suppress background noise while enhancing the weight of the oil spill area in SAR images. Furthermore, to achieve optimal segmentation performance along the boundaries of oil spill, the paper have included boundary measurements in the loss function. The results of experiment indicated that the proposed algorithm can achieve superior performance, with a pixel accuracy of 90.68%, a shortened number of parameters (only 1.52M), and an image inference speed of 28.53 milli-seconds, which surpasses other popular oil spill image segmentation models, such as Attention UNet, FCNet, PSPNet, and DeeplabV3 Net, in terms of both efficiency and quality of segmentation.
keywords:semantic segmentation of marine oil spill  lightweight  attention mechanism  loss function
查看全文   查看/发表评论   下载pdf阅读器