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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!