Joint Unsupervised Learning of Optical Flow and Depth by Watching Stereo Videos
arXiv:1810.03654
Abstract
Learning depth and optical flow via deep neural networks by watching videos has made significant progress recently. In this paper, we jointly solve the two tasks by exploiting the underlying geometric rules within stereo videos. Specifically, given two consecutive stereo image pairs from a video, we first estimate depth, camera ego-motion and optical flow from three neural networks. Then the whole scene is decomposed into moving foreground and static background by compar- ing the estimated optical flow and rigid flow derived from the depth and ego-motion. We propose a novel consistency loss to let the optical flow learn from the more accurate rigid flow in static regions. We also design a rigid alignment module which helps refine ego-motion estimation by using the estimated depth and optical flow. Experiments on the KITTI dataset show that our results significantly outperform other state-of- the-art algorithms. Source codes can be found at https: //github.com/baidu-research/UnDepthflow
arXiv commentSubmitted to AAAI 2019
References in corpus (8)
- Adam: A Method for Stochastic Optimization
- Unsupervised CNN for Single View Depth Estimation: Geometry to the Rescue
- Unsupervised Learning of Depth and Ego-Motion from Video
- PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume
- Unsupervised Learning of Geometry with Edge-aware Depth-Normal Consistency
- Unsupervised Learning of Monocular Depth Estimation and Visual Odometry with Deep Feature Reconstruction
- Pyramid Stereo Matching Network
- Every Pixel Counts: Unsupervised Geometry Learning with Holistic 3D Motion Understanding