onnx
洪流之源
这个作者很懒,什么都没留下…
展开
-
OpenVINO部署Yolov5
0.开发环境硬件平台:8 11th Gen Intel(R) Core(TM) i5-1145G7E @ 2.60GHz操作系统:Ubuntu 20.04 (LTS)OpenVINO: 2021.04(LTS)Python: 3.8.5 (conda 虚拟环境)Pytorch:1.9.0+cpuonnx:1.9.0onnx-simplifier: 0.3.6onnxruntime: 1.8.1onnxoptimizer: 0.2.61. yolov5转onnx...原创 2021-07-21 19:05:51 · 1536 阅读 · 1 评论 -
yolov5 onnx 推理代码
onnx_yolov5_inference.pyimport osimport cv2import timeimport onnxruntime as ortimport numpy as npfrom common import pad_to_squarefrom common import save_tensorfrom common import decodefrom common import non_max_suppressionfrom common import dra原创 2021-06-29 10:55:30 · 2342 阅读 · 1 评论 -
onnx模型修改、添加Node
import onnximport copy# onnx 插入新的Nodedef insert_node(model, insert_node, follow_up_node): # 根据插入Node的输出修改后续node的输入 follow_up_node.input[0] = insert_node.output[0] # 找到后续Node的索引位置,并将插入节点插入到graph中 for follow_up_node_index, _follow_up_nod.原创 2021-05-12 11:53:43 · 2784 阅读 · 0 评论 -
Retinaface Pytorch转ONNX
环境:Retinaface Pytorch源码:import timeimport torchimport torch.nn as nnimport torchvision.models._utils as _utilsimport torchvision.models as modelsimport torch.nn.functional as Ffrom torch.autograd import Variabledef conv_bn(inp, oup, stride原创 2021-04-25 15:44:29 · 484 阅读 · 0 评论 -
Pytorch实现CSPDarknet53
import torchimport torch.nn.functional as F import torch.nn as nnimport mathfrom collections import OrderedDictimport numpy as npimport torch.onnx# Mish激活函数: f(x) = x * tanh(log(1 + e^x))# pytorch中F.softplus(x)等同于torch.log(1 + torch.exp(x))class.原创 2021-04-18 22:01:58 · 1475 阅读 · 1 评论 -
Pytorch DataParallel多卡训练模型导出onnx模型
Pytorch模型转换到onnx模型代码如下:import torchimport torch.nn as nnimport torch.onnximport onnximport osfrom QualityNet import QualityNetif __name__ == '__main__': os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] =原创 2021-03-29 18:29:49 · 1998 阅读 · 2 评论