yolov5官方的yolov5s.pt文件可以识别80个类别,如何用OpenCV使用这个文件做目标识别

官方的yolov5s.pt文件可以识别80个类别,如何用OpenCV使用这个文件做目标识别?
目前已把pt文件转为onnx,但是不知道对不对,也不知道怎么使用,现需要用opencv使用yolov5的yolov5s.pt文件做目标识别的python代码,可以的话加点注释。

我有opencv的c++版本,你一定要python的opencv版本吗,如果是python的话,直接用torch就可以了,速度快,转用opencv反而速度慢。
如果一定要的话,直接用yolov5自带的detect.py文件,框中两个修改的地方,前面的onnx为官方的export.py运行yolov5s.pt导出的onnx模型,路径写好就行,最后的dnn这里加个参数,表示用opencv推理onnx。
如果你觉得需要单独的opencv的话,debug一下,把运行dnn过程中的相关代码抽取出来就可以了。举个例子来说,

img


debug一句一句运行,你会发现在这里就是用opencv读取onnx模型。

img