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) 
相关推荐
YRr YRr4 分钟前
深度学习:神经网络的搭建
人工智能·深度学习·神经网络
威桑6 分钟前
CMake + mingw + opencv
人工智能·opencv·计算机视觉
爱喝热水的呀哈喽10 分钟前
torch张量与函数表达式写法
人工智能·pytorch·深度学习
MessiGo19 分钟前
Python 爬虫 (1)基础 | 基础操作
开发语言·python
肥猪猪爸43 分钟前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
LZXCyrus1 小时前
【杂记】vLLM如何指定GPU单卡/多卡离线推理
人工智能·经验分享·python·深度学习·语言模型·llm·vllm
Enougme1 小时前
Appium常用的使用方法(一)
python·appium
懷淰メ1 小时前
PyQt飞机大战游戏(附下载地址)
开发语言·python·qt·游戏·pyqt·游戏开发·pyqt5
我感觉。1 小时前
【机器学习chp4】特征工程
人工智能·机器学习·主成分分析·特征工程
hummhumm2 小时前
第 22 章 - Go语言 测试与基准测试
java·大数据·开发语言·前端·python·golang·log4j