donut导出onnx模型

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) 
相关推荐
ARM+FPGA+AI工业主板定制专家8 分钟前
基于ZYNQ FPGA+AI+ARM 的卷积神经网络加速器设计
人工智能·fpga开发·cnn·无人机·rk3588
伏小白白白11 分钟前
【论文精度-1】 组合优化中的机器学习:方法论之旅(Yoshua Bengio, 2021)
人工智能·机器学习·组合优化
CH3_CH2_CHO17 分钟前
DAY03:【DL 第一弹】神经网络
人工智能·pytorch·深度学习·神经网络
dlraba80218 分钟前
用 Python+OpenCV 实现实时文档扫描:从摄像头捕捉到透视矫正全流程
开发语言·python·opencv
小熊出擊24 分钟前
【pytest】fixture 内省(Introspection)测试上下文
python·单元测试·pytest
算家计算32 分钟前
蚂蚁开源万亿参数大模型Ling-1T:多项能力全球领先
人工智能·开源·资讯
leijiwen33 分钟前
S11e Network 商业模型:AI × Web3 × RWA 驱动的实体经济新范式
人工智能·web3·区块链
njsgcs44 分钟前
sse mcp flask 开放mcp服务到内网
后端·python·flask·sse·mcp
一人の梅雨1 小时前
1688 店铺商品全量采集与智能分析:从接口调用到供应链数据挖掘
开发语言·python·php