9 lines
195 B
Python
9 lines
195 B
Python
from ultralytics import YOLO
|
|
|
|
|
|
'''
|
|
导出yolo模型并不能转换成tensorRT模型
|
|
'''
|
|
|
|
model = YOLO('5000_HeadHelmet/train2/weights/best.pt')
|
|
model.export(format="engine", device=0) |