FPGA Accelerated Real-time Recurrent All-Pairs Field Transforms for Optical Flow
Yingxiang Li, Yingke Gao, Zhiwen Su, Shitao Chen, Longjun Liu · 2022
Optical flow algorithms based on deep learning have achieved excellent performance on multiple datasets, bringing new opportunity for optical flow estimation. Recurrent All-Pairs Field Transforms (RAFT) is one of the most powerful deep network based optical flow algorithms, but it is difficult to process in real time on the resource-limited embedded platform. In this paper, we propose RAFT-Lite by compressing the original RAFT model, which is more lightweight and suitable for hardware deployment. We further propose a hardware accelerating architecture on FPGA for RAFT-Lite, which provides an efficient scheduling strategy for the convolution in RAFT to achieve efficient pipeline and resource reuse. On Xilinx ZCU102 evaluation board, the accelerated hardware system can reach 10.4fps processing images with a resolution of 512*396, which is 6.8x of [email protected] and 46x of ARM [email protected]. Besides, the power consumption is 13.103W.