问题:在训练完成后想在测试集上验证精度
解决方案
from ultralytics import YOLO
# Load a model
model = YOLO('yolov8m.pt') # load an official model
model = YOLO('runs/detect/yolov8_640_ep200/weights/best.pt') # load a custom model
# Validate the model
metrics = model.val(split='test') # no arguments needed, dataset and settings remembered
metrics.box.map # map50-95
metrics.box.map50 # map50
metrics.box.map75 # map75
metrics.box.maps # a list contains map50-95 of each category
在官方文档介绍中,可以得知split关键字可以选择验证时所需的数据集
默认为val,如果想要yaml文件中划分的测试集上查看模型的精度,仅需要将split字段更改即可。