Mobile real-time detection transformer for computationally constrained devices
摘要
The recent state-of-the-art model, Real Time Detection Transformer (RT-DETR), has surpassed YOLOv8 in terms of both accuracy and speed. We present a model based on the RT-DETR architecture, with a lightweight backbone, to achieve compactness and high speed. Our model, Mobile RT-DETR (MRTDETR), utilises a series of MobileNetv3 blocks. Specifically, the layers 7, 13, and 15 of the MobileNetv3 backbone were used as inputs to the encoder. Separable self attention was also explored, but not retained. MRTDETR achieves 44.6% mAP with a train batch size of 4 on NVIDIA GeForce RTX 3050 OEM GPU and 43.1% mAP with a train batch size of 64 on NVIDIA H100 PCIe GPUs, hence surpassing existing implementations of FasterRCNN, MobileFormer, and MobileViT. With 15.23 GFLOPs, MRTDETR comes close to the performance of the state-of-the-art YOLOv8 and YOLOv9 detectors of similar size while having