Puck tracking system for aerohockey game with YOLO2

Abstract

This article is devoted to the preparation of the YOLO2 convolutional neural network for use in the robot artificial intelligence tracking module in air hockey competitions in the human-robot or robot-robot category. Such developments are unique in the sphere of artificial intelligence and robotic technologies. The task was to collect and prepare educational material, as well as further training and testing of the convolutional neural network to create a "vision" and further predict the trajectory of a fixed object. For this, the YOLO2 model was used, created by the developer Ali Farhadi in the low-level language C [1]. As a result of testing in the next game, the results of the correct detection of the object in the range of 80% were obtained, despite the lack of a standard camera position and poor image quality. In the future, the created system can be used in the further development of the AI robot systems and for creating tactics of behavior in various air hockey matches.