OpenVINO与TensorRT部署YOLOv10对象检测

文章预览

点击上方 蓝字 关注我们 微信公众号： OpenCV学堂 关注获取更多计算机视觉与深度学习知识 模型导出与输入输出 YOLOv10模型导出ONNX脚本如下： from ultralytics import YOLOv10 """Test exporting the YOLO model to ONNX format.""" f = YOLOv10( "yolov10n.pt" ).export(format= "onnx" , opset= 11 , dynamic= False )  如果不指明opset=11或者12，默认opset=10导出的模型推理时候会非常的慢。指定opset=11以后，导出的模型结构如下： 输入与输出格式解释如下： 输入支持格式为 :1x3x640x640 输出格式为1 x300x6 输出格式300是指输出得预测框数目，6分别是 x1 y1 x2 y2 score classid C++推理 OpenVINO2023 C++ 推理 ，发现一个很大的BUG，就是当使用AUTO模式的时候： ov::CompiledModel compiled_model = ie.compile_model( "D:/python/yolov10-1.0/yolov10n.onnx" , "AUTO" ); auto infer_request = compiled_model.create_infer_request(); 模型推理针对视频推理会发生 ………………………………