YOLOv8训练好的pt文件如何用来预测

2023-12-13 07:20:44

1. 使用原版预测

代码如下:

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n.pt')  # load an official model

# Predict with the model
results = model('https://ultralytics.com/images/bus.jpg')  # predict on an image

命令如下:

yolo detect predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg'  # predict with official model

2.使用自己的模型

仅需要把自己的模型放入即可
代码如下:

from ultralytics import YOLO

# Load a model
model = YOLO('path/to/best.pt')  # load a custom model

# Predict with the model
results = model('https://ultralytics.com/images/bus.jpg')  # predict on an image

命令如下:

yolo detect predict model=path/to/best.pt source='https://ultralytics.com/images/bus.jpg'

会输出一下信息
在这里插入图片描述

3.如何让图片保存下来

添加save=True即可,代码如下:

from ultralytics import YOLO

# Load a model
model = YOLO('path/to/best.pt')  # load a custom model

# Predict with the model
results = model('https://ultralytics.com/images/bus.jpg', save=True)  # predict on an image

会提示保存的路径
在这里插入图片描述

文章来源:https://blog.csdn.net/qq_43471945/article/details/134675557
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。