Optical Flow Estimation Using Spatial-Channel Combinational Attention-Based Pyramid Networks

Xuezhi Xiang,Mingliang Zhai,Rongfang Zhang,Ning Lv,Abdulmotaleb El Saddik
DOI: https://doi.org/10.1109/icip.2019.8804417
2019-01-01
Abstract:Recently, learning to estimate optical flow via deep convolutional networks is attracting significant attention. In this paper, we introduce a spatial-channel attention module into optical flow estimation, which infers attention maps along two separated dimensions, channel and spatial, and then integrates these separated attention maps into a fusion attention map for feature refinement. We embed this module into spatial pyramid network, which can adaptively learn the channel and spatial attention maps at each level for modifying the different scaled features and can further improve the accuracy of optical flow estimation. Our network is trained on FlyingChairs and FlyingThings3D datasets with a supervised manner, and is further tested on MPI-Sintel benchmark. The experimental results show that using the spatial-channel attention unit is beneficial for dense flow estimation and our approach is comparable with the state-of-the-art methods.
What problem does this paper attempt to address?