15 lines
447 B
Python
15 lines
447 B
Python
from ultralytics import YOLO
|
||
|
||
'''
|
||
注意修改配置文件中的文件夹路径
|
||
'''
|
||
|
||
# train2，yolov8n
|
||
model = YOLO('5000_HeadHelmet/train/weights/best.pt')
|
||
|
||
# Validate the model
|
||
metrics = model.val() # no arguments needed, dataset and settings remembered
|
||
print(metrics.box.map) # map50-95
|
||
print(metrics.box.map50) # map50
|
||
print(metrics.box.map75) # map75
|
||
print(metrics.box.maps) # a list contains map50-95 of each category |