Non-uniformity seriously affects the imaging quality of infrared focal plane.To solve the problem,a non-uniformity correction algorithm is proposed based on the sub-pixel registration and the momentum-based BP neural network.Because of the shift of optical axis and imaging target location in the shortwave infrared camera,the sub-pixel registration algorithm based on matrix multiplication is used for registration.In order to accelerate the convergence of the algorithm,the two-point method is used to initialize the correction coefficient.As the BP neural network is easy to fall into the local optimal value,the correction effect is improved by increasing the momentum of the neural network method.The simulation results show that the proposed algorithm can eliminate the problems of ghost and edge blur in the traditional neural network correction method,obtain good correction effect and can improve the convergence speed of the algorithm,which provides a good foundation for short-wave infrared post-processing.