用opencv部署yolov5s.onnx调用摄像头时报错,报错信息如下:
在使用yolov5s.onnx会报错是因为在处理 ONNX 模型中的 Floor
节点时发生错误,具体来说是由于类型不匹配或尺寸不一致导致的,所以可以直接用onnxsim对onnx模型进行简化处理,把Floor层简化掉
1.安装onnxsim包
pip install onnx-simplifier
2.onnxsim简化模型
在生成onnx模型的文件夹下,打开终端
onnxsim yolov5s.onnx yolov5s_smp.onnx
可以看出,简化后的模型floor层已经没有了
用生成的yolov5s_smp.onnx文件就可以取代原来的onnx运行,并且不会报错
问题解决