摘要: onnx导出-多输入+动态维度 目录onnx导出-多输入+动态维度常见问题多参数输入动态输入导出动态输入问题-无法修改维度重新定义onnx输出验证导出和测试多头输入多头输出参考资料 常见问题 多参数输入 import numpy as np import cv2 import torch impor 阅读全文
posted @ 2024-01-29 21:18 贝壳里的星海 阅读(400) 评论(0) 推荐(0) 编辑
摘要: onnx模型导出 目录onnx模型导出环境准备简介介绍torch.onnx.export参数解析onnx导出步骤单输入导出示例定义并准备模型加载权重并测试onnx导出和验证Netron可视化在线使用代码可视化onnx模型推理补充细节添加自定义标签读取自定义标签导出注意参考资料 环境准备 # 环境依赖 阅读全文
posted @ 2024-01-29 21:18 贝壳里的星海 阅读(197) 评论(0) 推荐(0) 编辑