1. RT-DETR
RT-DETR是由由此,百度推出了——RT-DETR (Real-Time DEtection TRansformer) ,一種基于 DETR 架構的實時端到端檢測器,其在速度和精度上取得了 SOTA 性能。
RT-DETR開源的代碼在百度自己的飛槳paddlepaddle上,因此非常不便于我們使用。以下介紹一下YOLOv8集成的RT-DETR如何使用。
2. YOLOv8集成的RT-DETR
2.1 下載權重文件
下載官方提供的模型:rtdetr-l.pt
2.2 如何使用
模型結構如下圖位置所示:即ultralytics/cfg/models/rt-detr/rtdetr-l.yaml
YOLOv8集成了RT-DETR的l與x模型
使用Python代碼可以使用以下代碼:
from ultralytics import RTDETR# Load a COCO-pretrained RT-DETR-l model
model = RTDETR('rtdetr-l.pt')# Display model information (optional)
model.info()# Train the model on the COCO8 example dataset for 100 epochs
results = model.train(data='coco8.yaml', epochs=100, imgsz=640)# Run inference with the RT-DETR-l model on the 'bus.jpg' image
results = model('path/to/bus.jpg')
使用命令,可以如下:
第一行是訓練,第二行是預測
# Load a COCO-pretrained RT-DETR-l model and train it on the COCO8 example dataset for 100 epochs
yolo train model=rtdetr-l.pt data=coco8.yaml epochs=100 imgsz=640# Load a COCO-pretrained RT-DETR-l model and run inference on the 'bus.jpg' image
yolo predict model=rtdetr-l.pt source=path/to/bus.jpg