基于改进SOLOv2算法的海滩影像水边线提取研究

Research on waterline extraction from beach images based on improved SOLOv2 algorithm

  • 摘要: 海滩水边线的提取是海洋科学和海岸工程领域的一项重要研究任务。为缓解传统方法在处理该任务时存在的噪声敏感、阈值不稳定、需手动调参等问题,本文对影像数据进行了极值预处理,并提出了基于可变形卷积的改进SOLOv2水边线提取算法。本文探讨了特征提取网络深度和可变形卷积(DCN)应用层数对算法精度与速度的影响。随着特征提取网络深度的增加,水边线提取算法的精度逐渐提高,但速度明显降低,实际特征提取网络深度以50为宜;随着DCN应用层数的增加,水边线提取算法的精度呈先增加后减小的趋势,实际应用层数以2为宜;相较于原始SOLOv2算法,改进SOLOv2算法在平均精度(average precision, AP)和平均交并比(intersection over union, IoU)上的性能分别提升了1.7%和0.1%。此外,通过将算法权重由单精度转换为半精度,进一步提升了算法的推理速度。

     

    Abstract: The extraction of beach waterline is an important research task in the field of marine science and coastal engineering. In order to alleviate the problems of noise sensitivity, unstable thresholds, and manual parameter adjustment that exist in traditional methods when dealing with this task, this paper performs extreme value preprocessing on image data and proposes an improved SOLOv2 waterline extraction algorithm based on deformable convolution. This paper explores the effects of feature extraction network depth and the number of deformable convolution (DCN) application layers on the accuracy and speed of the algorithm. As the depth of the feature extraction network increases, the accuracy of the waterline extraction algorithm gradually improves, but the speed decreases significantly. The actual depth of the feature extraction network is recommended to be 50; As the number of DCN application layers increases, the accuracy of the waterline extraction algorithm shows a trend of first increasing and then decreasing. The actual number of the application layers is recommended to be 2; Compared with the original SOLOv2 algorithm, the improved SOLOv2 algorithm has improved performance by 1.7% and 0.1% in average precision (AP) and average Intersection over Union (IoU), respectively. In addition, by converting the algorithm weight from single precision to half precision, the inference speed of the algorithm has been further improved.

     

/

返回文章
返回