视觉检测大模型DRTR部署使用-Yolo
·
视觉检测大模型DRTR部署使用-Yolo
from ultralytics import RTDETR
import cv2
# 1. 加载预训练模型（支持 rtdetr-l.pt, rtdetr-x.pt 等）
model = RTDETR("rtdetr-l.pt") # 首次运行会自动下载权重
# 2. 单张图片推理
results = model("http://images.cocodataset.org/val2017/000000039769.jpg")
# 3. 显示结果
results[0].show() # 直接显示图片
# 或者获取检测结果数据
boxes = results[0].boxes # 检测框对象
for box in boxes:
print(f"类别: {model.names[int(box.cls)]}, 置信度: {box.conf:.3f}, 坐标: {box.xyxy.tolist()}")
更多推荐
0
0- 0
已为社区贡献25条内容
所有评论(0)