Giter Club home page Giter Club logo

yolov5-v6.1-opencv-onnxrun's Introduction

yolov5-v6.1-opencv-onnxrun

看到https://github.com/ultralytics/yolov5 在最近更新的v6.1版本的,我编写了一套

使用OpenCV部署yolov5-v6.1目标检测,包含C++和Python两个版本的程序

使用ONNXRuntime部署yolov5-v6.1目标检测,包含C++和Python两个版本的程序

支持yolov5s,yolov5m,yolov5l,yolov5n,yolov5x, yolov5s6,yolov5m6,yolov5l6,yolov5n6,yolov5x6的十种结构的yolov5-v6.1

由于模型文件太多,我没有直接上传到仓库里,模型文件存放在百度云盘里

下载链接:https://pan.baidu.com/s/1pz1A4YOurxL5G0_ITVEerg 提取码:zm2q

转换生成onnx文件的方法,可以参考我的csdn博客文章 https://blog.csdn.net/nihate/article/details/112731327 或者知乎文章 https://zhuanlan.zhihu.com/p/466677699

yolov5-v6.1-opencv-onnxrun's People

Contributors

hpc203 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

yolov5-v6.1-opencv-onnxrun's Issues

opencv版本问题

大佬你好,感谢你的优秀工作!!我想问一下这个项目对于opencv版本有要求吗?不知道在4.1上能否顺利运行?

不画Bounding Box (出现负数)

今天在测试代码的时候发现,结果没有框。通过print log发现bbox的坐标很大。e.g. [45355, 43453, -43453, -34343]

应该是一个bug出现在postprocess之前,参与anchor那里。我后来把anchor那一部分全部comment掉,运行没有问题。

但我想请教一下anchor,grid在这里起到什么作用?

请大佬说明下各个模块的版本

下载debug发现运行有很多版本不兼容问题,很打击信心。为了较少不必要的问题发生,有请大佬 把自己的OPENCV ,PYHON 等所有模块的信息公布出来参考,增加易用性,我想star会猛增的!

yolov5v5推理结果解析问题请教

我看结果都是取到数据后,直接取前四个就是cx cy w h:
图片
但是我看您的代码里面确是:
float cx = (pdata[0] * 2.f - 0.5f + j) * stride; ///cx
float cy = (pdata[1] * 2.f - 0.5f + i) * stride; ///cy
float w = powf(pdata[2] * 2.f, 2.f) * anchor_w; ///w
float h = powf(pdata[3] * 2.f, 2.f) * anchor_h; ///h

不知道怎么理解?

ort_session->Run报段错误

请问onnxruntime版本是多少呢,ort_session->Run段错误是否也跟官方导出onnx与代码中使用的不一致有关呢,谢谢

main.cpp 进行推理闪退

您好,我用qt5.7.1 和 opencv4.5.3 使用您的 main.cpp 进行推理,加载是您百度的模型权重,程序直接闪退。请问您知道是什么原因吗?

win10 cpp opencv 推理无法运行

环境:win10、VS2019、OpenCV4.5.4
使用提供的模型yolov5n.onnx,运行opencv下的main.cpp

报错Expression:vector subscript out of range

debug模型加载、forword都没问题。

初步判断是在detect的这两行报错,应该是网络输出结果有问题?

int num_proposal = outs[0].size[1];
int nout = outs[0].size[2];

我打印outs
cout << "outs size " << outs.size() << " " << outs[0].cols << " " << outs[0].rows << endl;

得到的结果是1 -1 -1 不知道是否正常?

some error

现在yolov5导出来的模型都将三个检测框的结果合并到一个维度里面去了,为1x25200x38,不再是单独的了,代码可能得改改了

GPU推理

请问,如何进行GPU加速推理呢,我看程序是通过CPU推理的

yolov5使用onnxruntime进行多图推理

示例代码只是检测一张图片,请问如果要一次性输入一个batch的图片,可以进行多图推理吗?我在导出onnx的时候导出了动态batch的onnx模型,但是就无法推理了,请问这部分的代码大概需要怎么修改呢

readNet 加载onnx

我用您的代码加载从网盘里下载的yolov5s.onnx。readNet处报错:Microsoft C++ 异常: std::bad_alloc,位于内存位置 0x00000055455CEC60 处

python opencv加载onnx

程序跑通了 调试代码也可以看到具体检测 person tie
但是最后出图 没有框,想请教下是哪里出了问题

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.