donut模型简介
代码
onnx导出
shell
# 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
模型结构查看
netron
模型太大无法上传,放弃
onnx查看网络结构
python
#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)