donut导出onnx模型

donut模型简介

代码

onnx导出

# pip install optimum-cli
# 导出为encoder.onnx 和 decoder.onnx
optimum-cli export onnx  -m finetune-model-path --task vision2seq-lm onnx-ouput-path 

#  导出为model.onnx模型
optimum-cli export onnx  -m finetune-model-path --task vision2seq-lm onnx-ouput-path --monolith
1
2
3
4
5
6

模型结构查看

netron

模型太大无法上传，放弃

onnx查看网络结构

#https://zhuanlan.zhihu.com/p/516920606
import onnx 
model = onnx.load('model.onnx') 
# 模型图结构
graph = model.graph 
# 模型节点
node = graph.node 
# 模型输入
input = graph.input 
# 模型输出
output = graph.output 
print(node) 
print(input) 
print(output) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14