yolo进行视频检测结果没有生成

最新推荐文章于 2024-09-09 09:54:33 发布

暗巷提灯

最新推荐文章于 2024-09-09 09:54:33 发布

阅读量184

点赞数 2

文章标签： YOLO python 人工智能

本文链接：https://blog.csdn.net/m0_75015083/article/details/138921621

版权

你可能用了这套代码：

from ultralytics import YOLO

# Load a pretrained YOLOv8n model
model = YOLO('./best.pt')

# Define path to video file
source = r".\WeChat_20240515193007.mp4"

# Run inference on the source
results = model(source, stream=True)  # generator of Results objects

这套代码它把结果放到了results中但没有保存。

正确代码：


from ultralytics import YOLO

# Load a pretrained YOLOv8n model
model = YOLO(r"./best.pt")

# Run inference on 'bus.jpg' with arguments
model.predict(r"./p1_rgb.mp4", save=True, imgsz=320, conf=0.5)