from ultralytics import YOLO


'''
    导出yolo模型并不能转换成tensorRT模型
'''

model = YOLO('5000_HeadHelmet/train2/weights/best.pt')
model.export(format="engine", device=0)