search

hpc203 / yolov5-face-landmarks-opencv

yolov5检测人脸和关键点，只依赖opencv库就可以运行，程序包含C++和Python两个版本的
62 stars 15 forks source link

模型转换 #1

Open cLEARLoVE00 opened 3 years ago

cLEARLoVE00 commented 3 years ago

作者你好，我在使用torch1.7环境转换的onnx模型，在使用你的脚本调用的时候onnx做预测的时候出现cv2.dnn.readNet(‘onnx路径’)提示 blob.type()==5 in function‘populateNet’，想问下你有没有这种情况，如何解决的？谢谢

hpc203 commented 3 years ago

作者你好，我在使用torch1.7环境转换的onnx模型，在使用你的脚本调用的时候onnx做预测的时候出现cv2.dnn.readNet(‘onnx路径’)提示 blob.type()==5 in function‘populateNet’，想问下你有没有这种情况，如何解决的？谢谢

看我的csdn博客文章https://blog.csdn.net/nihate/article/details/112731327 ，在转换生成onnx文件时需要注意的事项。检查一下你在转换生成模型时的model，在model里的Detect类是不是包含了decode

cLEARLoVE00 commented 3 years ago

我用了你官方转好的onnx模型在做测试的时候也会出现上述这个错误，想了解具体的测试流程

hpc203 commented 3 years ago

我用了你官方转好的onnx模型在做测试的时候也会出现上述这个错误，想了解具体的测试流程 你的python-opencv的版本是多少的

cLEARLoVE00 commented 3 years ago

版本是4.1.2

hpc203 commented 3 years ago

版本是4.1.2

你升级opencv到4.5版本以上的再运行程序试试看

JinXiangLai commented 3 years ago

版本是4.1.2

你升级opencv到4.5版本以上的再运行程序试试看

4.5.2 UBUNTU18 ,有一样的问题