6 lines
268 B
Python
6 lines
268 B
Python
|
#更多功能参数介绍：https://docs.ultralytics.com/modes/export/#key-features-of-export-mode
|
|||
|
|
from ultralytics import YOLO
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
model = YOLO('best.pt') # load a custom trained model
|
|||
|
|
# Export the model
|
|||
|
|
model.export(format='onnx')
|