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)

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐