Object tracking algorithm for unmanned surface vehicle based on improved mean-shift method

Abstract

The unmanned surface vehicle has the characteristics of high maneuverability and flexibility. Object detection and tracking skills are required to improve the ability of unmanned surface vehicle to avoid collisions and detect targets on the surface of the water. Mean-shift algorithm is a classic target tracking algorithm, but it may fail when pixel interference and occlusion occur. This article proposes a tracking algorithm for unmanned surface vehicle based on an improved mean-shift optimization algorithm. The method uses the self-organizing feature map spatial topology to reduce the interference of the background pixels on the target object and predicts the center position of the object when the target is heavily occluded according to the extended Kalman filter. First, a self-organizing feature map model is built to classify pixels in a rectangular frame and the background pixels are extracted. Then, the method optimizes the extended Kalman filter solution process to complete the prediction and correction of the target center position and introduces a similarity function to determine the target occlusion. Finally, numerical analyses based on a ship model sailing experiment are performed with the help of OpenCV library. The experimental results validated that the proposed method significantly reduces the cumulative error in the tracking process and effectively predicts the position of the target between continuous frames when temporary occlusion occurs. The research can be used for target detection and autonomous navigation of unmanned surface vehicle.