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