Wangweilai1 / yolov7-opencv-onnxrun-cpp-py

分别使用OpenCV、ONNXRuntime部署YOLOV7目标检测,一共包含14个onnx模型,依然是包含C++和Python两个版本的程序

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

yolov7-opencv-onnxrun-cpp-py

分别使用OpenCV、ONNXRuntime部署YOLOV7目标检测,一共包含14个onnx模型,依然是包含C++和Python两个版本的程序。

编写这套YOLOV7的程序,跟此前编写的YOLOV6的程序,大部分源码是相同的,区别仅仅在于图片预处理的过程不一样。 YOLOV7的图片预处理是BGR2RGB+不保持高宽比的resize+除以255

由于onnx文件太多,无法直接上传到仓库里,需要从百度云盘下载, 链接: https://pan.baidu.com/s/1FoC0n7qMz4Fz0RtDGpI6xQ 密码: 7mhs ,下载完成后把models目录放在主程序文件的目录内,编译运行

使用opencv部署的程序,有一个待优化的问题。onnxruntime读取.onnx文件可以获得输入张量的形状信息, 但是opencv的dnn模块读取.onnx文件无法获得输入张量的形状信息,目前是根据.onnx文件的名称来解析字符串获得输入张量的高度和宽度的。

YOLOV7的训练源码是 https://github.com/WongKinYiu/yolov7 跟YOLOR是同一个作者的。

About

分别使用OpenCV、ONNXRuntime部署YOLOV7目标检测,一共包含14个onnx模型,依然是包含C++和Python两个版本的程序


Languages

Language:Python 52.9%Language:C++ 47.1%