OpenCV加载onnx模型实现SSD,YOLOV3,YOLOV5的推理

目前SSD和YOLO是工业界使用最多的两种检测器。最近整理了一下自己实现的SSD,YOLOV3和YOLOV5推理代码,项目虽然是基于OpenCV实现的,但是由于使用的是onnx模型,所以代码可以很容易移植到TensorRT,NCNN等其他C++推理框架。在实现的过程中基本是参考了原作者的开源代码,也参考了网上的一些代码,由于记不清出处,这里无法给出参考代码的地址,这里感谢各位作者。希望自己的工作能够帮助更多初学者学习SSD和YOLO。项目地址:https://github.com/qianqing13579/DetectorSamples,欢迎star。有什么问题可以留言讨论。

你可能感兴趣的:(目标检测,OpenCV,pytorch,神经网络,自动驾驶)