YOLO V8 predict.py将命令行运行改为直接运行

最新推荐文章于 2025-06-17 18:13:58 发布

方方爱学习

最新推荐文章于 2025-06-17 18:13:58 发布

阅读量1.7k

点赞数 11

CC 4.0 BY-SA版权

分类专栏： YOLO系列笔记文章标签： YOLO 深度学习

本文链接：https://blog.csdn.net/weixin_45920120/article/details/135626879

文章讲述了如何在YOLOv8中通过命令行快速运行预测代码，提供必要的参数设置和代码示例，方便初学者调试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

YOLOv8命令行运行代码为：yolo predict model=yolov8n.pt source='ultralytics/assets/bus.jpg'

在入门阶段进行调试时每次需在命令行运行较为麻烦，修改主函数参数的调用

主要修改的地方如下：

传入所需运行的参数模型 model

传入所需预测的图片地址 source

代码如下：

# Ultralytics YOLO 🚀, AGPL-3.0 license

import torch
import ultralytics
# from ultralytics import YOLO


from ultralytics.engine.predictor import BasePredictor
from ultralytics.engine.results import Results
from ultralytics.utils import DEFAULT_CFG, ROOT, ops


class DetectionPredictor(BasePredictor):

    def postprocess(self, preds, img, orig_imgs):
        """Postprocesses predictions and returns a list of Results objects."""
        preds = ops.non_max_suppression(preds,
                                        self.args.conf,
                                        self.args.iou,
                                        agnostic=self.args.agnostic_nms,
                                       max_det=self.args.max_det,