- 博客(7)
- 收藏
- 关注
RSS订阅原创 YOlOv8 ONNXRunTime-gpu 推理
(You Only Look Once）是一种流行的物体检测和图像分割模型，由华盛顿大学的约瑟夫-雷德蒙（Joseph Redmon）和阿里-法哈迪（Ali Farhadi）开发。显示检测结果，获取检测目标框，输出检测框，打印检测标签，保存检测结果为.bmp（其它格式文件会降低算法执行保存速率，其它格式需要系统会进行压缩，影响执行速率）根据YOLOv8官方文档介绍，通过使用model.export()将.pt文件转为.onnx文件。将符合检测要求的目标框，添加至预测坐标中。将检测模型转到GPU中推理。
2024-05-14 11:50:31
698
1
原创 yolov5 onnx推理报错
onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: images for the following indices index: 0 Got: 1 Expected: 16 Please fix either the inputs or the model.
2023-11-08 11:33:49
2106
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人