Design and Implementation of Automatic Assisted Aiming System For Robomaster EP Based on YOLOv5

Junjia Qin,Kangli Xu
2024-07-24
Abstract:In the crucial stages of the Robomaster Youth Championship, the Robomaster EP Robot must operate exclusively on autonomous algorithms to remain competitive. Target recognition and automatic assisted aiming are indispensable for the EP robot. In this study, we use YOLOv5 for multi-object detection to identify the Robomaster EP Robot and its armor. Additionally, we integrate the DeepSORT algorithm for vehicle identification and tracking. As a result, we introduce a refined YOLOv5-based system that allows the robot to recognize and aim at multiple targets simultaneously. To ensure precise tracking, we use a PID controller with Feedforward Enhancement and an FIR controller paired with a Kalman filter. This setup enables quick gimbal movement towards the target and predicts its next position, optimizing potential damage during motion. Our proposed system enhances the robot's accuracy in targeting armor, improving its competitive performance.
Robotics
What problem does this paper attempt to address?
The paper primarily focuses on the design and implementation of an Automatic Assisted Aiming System (AAA-System) for the RoboMaster Youth Championship. The core issue of the research is to enhance the robot's ability to identify and aim at targets during the competition, particularly improving its accuracy in dynamic environments. Specifically, the research addresses the following key issues: 1. **Multi-target Detection**: Using the YOLOv5 model for multi-target detection to identify RoboMaster EP robots and their armor plates. YOLOv5 was chosen for its efficient handling of high-density image data, making it particularly suitable for classification tasks in robotic tactical combat. 2. **Target Tracking**: Integrating the DeepSORT algorithm for vehicle identification and tracking, maintaining the unique identification of targets even in cases of occlusion or camera angle changes. By combining the Hungarian algorithm, it tracks strike data in real-time and determines the optimal shooting sequence. 3. **Prediction and Control**: Utilizing Kalman filters, PID controllers with feedforward enhancement, and FIR controllers to predict target trajectories and precisely control the robot's gimbal movement. These techniques optimize potential damage during shooting and ensure accuracy on fast-moving targets. 4. **Hardware Optimization**: Optimizing the YOLOv5 model by using stride convolution instead of max-pooling layers to improve the efficiency of small target recognition. Additionally, the deployment performance of the model on different hardware platforms (such as NVIDIA Jetson Nano, NVIDIA RTX 2080, etc.) was evaluated. In summary, this paper aims to develop a YOLOv5-based automatic assisted aiming system that enables RoboMaster EP robots to more accurately identify and aim at multiple targets during competitions, thereby enhancing their competitiveness.