ONNX
关注
分享
扫一扫
onnx相关工作，主要集中在模型的修改、转换及推理等
牛andmore牛
深度学习ing，人嘛，总要有点追求，要不和咸鱼有什么区别。。。。全部开源，用爱发光。
展开
-
onnxruntime-gpu与cuda版本对应及是否能调用cuda测试
onnxruntime-gpu安装及cuda版本对应关系和是否可以调用gpu测试原创 2025-03-11 16:21:55 · 2219 阅读 · 2 评论 -
onnx模型截取部分
onnx模型截取其中一部分原创 2024-05-11 14:44:32 · 751 阅读 · 0 评论 -
onnx模型动态输入形状转静态
onnx模型输入节点动态转静态原创 2024-05-11 14:33:14 · 1267 阅读 · 0 评论 -
获取onnx模型中权重并画出分布图
获取onnx模型中的权重参数并绘画直方图，从而查看模型参数分布原创 2023-05-04 09:40:11 · 1440 阅读 · 0 评论 -
3、picodet c++版onnxruntime推理及reshape和transpose的c++实现
k是每个检测头最后的大小，如输入是256，每个头的stride分别是[8,16,32,64],那么应有k就是[32,16,8,4],直接c++实现,有些搞不明白，先用python来实现。k ，这个m就是冒号这里的位置，取m=0,1,2,3,4,5,6,7,分别表示这一维度下的8个。关于c代码，可以查看我上边的PicoDet::generate_proposal里代码。4x8xkxk对应的是python: k,:,i,j，冒号就是全取，c对应的是：k。8,这是8个数的首地址，接差连取8个即可。原创 2023-04-19 14:51:13 · 844 阅读 · 0 评论 -
2、picodet转onnx裁剪及python onnxruntime推理
picodet转换onnx模型并对相关结构进行裁剪，最后进行推理原创 2023-04-19 10:10:22 · 1399 阅读 · 2 评论 -
onnx模型推理及获得各层输出结果
onnx指定gpu或cpu来进行推理，并获得各节点输出原创 2022-11-30 15:34:45 · 5023 阅读 · 0 评论 -
onnx模型显示输出形状
onnx模型显示形状原创 2022-11-30 14:45:21 · 2013 阅读 · 0 评论 -
onnx删除无用属性
onnx 删除node的属性原创 2022-11-30 11:06:31 · 1052 阅读 · 0 评论