onnx模型导出
【代码】onnx模型导出。
·

from ultralytics import YOLO
if __name__ == '__main__':
# model = YOLO("./weights/bestliver18.pt")
# model = YOLO("./weights/yolov8n.pt")
# model = YOLO("./weights/best.pt")
# model = YOLO("./weights/yolov8n-seg.pt")
# model = YOLO("./weights/yolov8n.pt")
# model = YOLO("./weights/yolov8n-cls.pt")
# model = YOLO("./weights/yolov8n-pose.pt")
model = YOLO(r"D:\已训练数据集\中药分类222种\yolov8n-best-cls.pt")
model.export(format='onnx', opset=12, batch=1, simplify=True)
更多推荐



所有评论(0)