基于 PWC-Net 的多层权值和轻量化 改进光流估计算法.

To solve the problem of insufficient real-time performance of existing optical flow estimation methods, this paper proposed a lightweight improved PSC-NET model with deep separable convolution ( depth pyramid, warping and cost volume, DS-PWC). One of the improvements was decoupling the conventional two-dimensional convolutional network layer into a deep separable convolutional layer. The other one was adding a weight coefficient based on the number of layers in the pyramid layer by DS-PWC, which greatly reduced the number of model parameters in the network structure without loss of precision. In addition, in the training process, the paper applied data enhancement technologies such as I+ ORE to further improve the generalization ability of estimation and prediction results. The experimental results show that the DS-PWC model was tested in the dataset and the operating efficiency reaches about 58 fps while maintaining the quality. To verify the effectiveness of the algorithm, this paper carried out the ablation experiments of model structure and data enhancement. The results verify the effectiveness of the DS-PWC model. [ABSTRACT FROM AUTHOR]