目录
十、YOLO-V8模型测试验证
10.1、YOLO-V8模型测试验证概述
YOLOv8模型训练结束后,测试模型的方式主要有两种:一种是通过命令行进行测试,另一种是使用Python API进行测试。
1. 通过命令行进行测试:可以使用如下的命令行参数来进行测试:`--weights`表示加载模型权重文件,`--source`表示输入数据文件或视频的路径,`--conf-thres`代表置信度阈值,即只有当检测到的目标的置信度大于这个阈值时,才会被视为真正的目标。例如:`python detect.py --weights yolov8n.pt --source 0.mp4 --conf-thres 0.25`。
2. 使用Python API进行测试:首先需要导入所需的库,然后实例化YOLO类并加载模型权重文件,最后调用detect方法来检测输入图像或视频。例如:`import torch from PIL import Image from ultralytics import YOLO model = YOLO("yolov8n.pt") results = model(image="test.jpg")`。
以上两种方式均可以完成对YOLOv8模型的测试,