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

Question

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

Answer 1

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

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