Object detection is a computer vision technique aimed at detecting and identifying objects in images or videos. In recent years, with advancements in Machine Learning and Deep Learning, object detection has made significant progress in various fields such as healthcare, security, and transportation. The DETR algorithm is a novel approach in object detection that combines transformer architecture with attention techniques to address object detection challenges. This research applies the DETR algorithm with ResNet backbone for vehicle detection on the roads, involving 6 object classes: Car, Truck, Bus, Motorcycle, Pickup Car, and Truck Box. Four training experiments were conducted: DETR-ResNet50, DETR-ResNet101, DETR-DC5-ResNet50, and DETR-DC5-ResNet101. The implementation results show that DETR-DC5 improves the accuracy of vehicle detection. DETR-DC5 with ResNet-101 achieved the highest score for AP50, which is 0.957. However, it should be noted that DETR-DC5 with ResNet-50 managed to maintain overall AP stability, with a lower parameter of 35.5. The model's outcomes in this study can be effectively applied for vehicle detection on the roads.
Copyrights © 2024