Onnxruntime之yolov7部署(cpp)

请添加图片描述

生成Onnx模型

首先 git clone https://github.com/WongKinYiu/yolov7
然后 cd ${yolov7 project root}
在当前路径下新建export_onnx.py

import torch
import torch.nn as nn
from torch.nn import Upsample

猜你喜欢

转载自blog.csdn.net/weixin_42990464/article/details/127407011