onnx模型裁剪一部分

最新推荐文章于 2024-07-24 17:34:51 发布

jstzwjr

最新推荐文章于 2024-07-24 17:34:51 发布

阅读量2.1k

点赞数

文章标签：深度学习人工智能

原文链接：https://github.com/onnx/onnx/blob/main/docs/PythonAPIOverview.md

版权

https://github.com/onnx/onnx/blob/main/docs/PythonAPIOverview.md

import onnx

input_path = 'path/to/the/original/model.onnx'
output_path = 'path/to/save/the/extracted/model.onnx'
input_names = ['input_0', 'input_1', 'input_2']
output_names = ['output_0', 'output_1']

onnx.utils.extract_model(input_path, output_path, input_names, output_names)