arrow
Return

WRD-Net: Water Reflection Detection using a parallel attention transformer

delete2024-08-01
delete0
PRE
AI
H
Huijie Dong
H
Hao Qi
H
Huiyu Zhou
J
Junyu Dong
X
Xinghui Dong *
DOI:10.1016/j.patcog.2024.110467delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In contrast to symmetry detection, Water Reflection Detection (WRD) is less studied. We treat this topic as a Symmetry Axis Point Prediction task which outputs a set of points by implicitly learning Gaussian heat maps and explicitly learning numerical coordinates. We first collect a new data set, namely, the Water Reflection Scene Data Set (WRSD). Then, we introduce a novel Water Reflection Detection Network, i.e., WRD-Net. This network is built on top of a series of Parallel Attention Vision Transformer blocks with the Atrous Spatial Pyramid (ASP-PAViT) that we deliberately design. Each block captures both the local and global features at multiple scales. To our knowledge, neither the WRSD nor the WRD-Net has been used for water reflection detection before. To derive the axis of symmetry, we perform Principal Component Analysis (PCA) on the points predicted. Experimental results show that the WRD-Net outperforms its counterparts and achieves the true positive rate of 0.823 compared with the human annotation.
Keywords:
Water Reflection Detection
Symmetry detection
Line detection
Multi-scale deep networks
Parallel attention transformer

Journal

Pattern Recognition cover
Pattern Recognition
IF:
7.6
Papers:
1.3W
Citations:
4.5W

Organization

O
ocean university of china
Scholars:
3.1W
Papers: 2.0W
Citations: 21
U
university of leicester
Scholars:
2.0W
Papers: 1.7W
Citations: 25