FDFlowNet: Fast Optical Flow Estimation using a Deep Lightweight Network
About
Significant progress has been made for estimating optical flow using deep neural networks. Advanced deep models achieve accurate flow estimation often with a considerable computation complexity and time-consuming training processes. In this work, we present a lightweight yet effective model for real-time optical flow estimation, termed FDFlowNet (fast deep flownet). We achieve better or similar accuracy on the challenging KITTI and Sintel benchmarks while being about 2 times faster than PWC-Net. This is achieved by a carefully-designed structure and newly proposed components. We first introduce an U-shape network for constructing multi-scale feature which benefits upper levels with global receptive field compared with pyramid network. In each scale, a partial fully connected structure with dilated convolution is proposed for flow estimation that obtains a good balance among speed, accuracy and number of parameters compared with sequential connected and dense connected structures. Experiments demonstrate that our model achieves state-of-the-art performance while being fast and lightweight.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Optical Flow Estimation | KITTI 2015 (train) | Fl-epe1.56 | 431 | |
| Optical Flow Estimation | MPI Sintel Final (train) | Endpoint Error (EPE)1.93 | 209 | |
| Optical Flow Estimation | MPI Sintel Clean (train) | EPE1.8 | 202 | |
| Optical Flow | MPI Sintel Clean (test) | -- | 158 | |
| Optical Flow | KITTI 2012 (train) | -- | 115 | |
| Optical Flow Estimation | KITTI 2015 (test) | Fl-all9.38 | 91 | |
| Optical Flow Estimation | KITTI 2012 (test) | AEE1.5 | 41 | |
| Optical Flow Estimation | MPI-Sintel final (test) | AEE5.11 | 30 | |
| Optical Flow | Sintel | Parameters (M)5.79 | 6 |