佟剑峰, 于雨, 韩少彬. 基于改进Mask R-CNN网络模型的水面物体检测算法[J]. 齐鲁工业大学学报, 2023, 37(5): 12-18. https://journal.qlu.edu.cn/CN/10.16442/j.cnki.qlgydxxb.2023.05.002.

摘要

随着深度学习的发展,利用深度学习相关知识实现水面物体检测的信息化以及智能化具有重要的意义。受到水平面复杂环境的影响,水面物体的实例分割依然是如今的难点之一。以水面物体实例分割为研究对象,提出了一种改进的Mask R-CNN网络模型。在数据集的构建阶段,使用labelme标注工具对图像进行标注并且通过添加高斯噪声、随机亮度等方法实现图像的增强。对于Mask R-CNN网络模型的改进部分,采用ResNeXt50作为骨干网络,并且在FPN网络中引入注意力机制SENet模块。实验结果表明,在IOU=0.5的判别方式中,改进后的网络模型相较于原始的Mask R-CNN网络模型,检测以及分割的平均精度分别提高到89.3%、88.5%。因此,改进后的Mask R-CNN网络模型相较于原始的Mask R-CNN网络模型更适合于水面物体的实例分割。