Git Product home page Git Product logo

Comments (7)

we0091234 avatar we0091234 commented on May 29, 2024

感谢作者提供的车牌检测的代码,请问作者能不能分享一下用yolv7实现的车牌检测转成onnx模型,我尝试着使用您在yolov5实现车牌检测中,将权重转成onnx,但在进行onnxruntime推理时,出现以下索引错误,发生异常: IndexError index 4 is out of bounds for axis 1 with size 3
python models/export.py --weights weights/plate_detect.pt --grid

from yolov7_plate.

Ellsionjeep avatar Ellsionjeep commented on May 29, 2024

感谢作者提供的车牌检测的代码,请问作者能不能分享一下用yolv7实现的车牌检测转成onnx模型,我尝试着使用您在yolov5实现车牌检测中,将权重转成onnx,但在进行onnxruntime推理时,出现以下索引错误,发生异常: IndexError index 4 is out of bounds for axis 1 with size 3
python models/export.py --weights weights/plate_detect.pt --grid

好的,非常感谢作者的提供,大力支持

from yolov7_plate.

Ellsionjeep avatar Ellsionjeep commented on May 29, 2024

再次打扰作者一下,我用您给的权重导出onnx后,在yolo_pose_onnx_inference.py的post_process函数中出现The truth value of an array with more than one element is ambiguous. Use a.any() or a.all(),我打印了det_scores[idx].shape 输出是(19,)这一部分维度是否有问题

from yolov7_plate.

we0091234 avatar we0091234 commented on May 29, 2024

再次打扰作者一下,我用您给的权重导出onnx后,在yolo_pose_onnx_inference.py的post_process函数中出现The truth value of an array with more than one element is ambiguous. Use a.any() or a.all(),我打印了det_scores[idx].shape 输出是(19,)这一部分维度是否有问题

onnx 推理的代码我还没跑过,19是没问题的,因为 xywh score这就是5个了,然后有两类2个分类得分,4个关键点,每个关键点包括x,y,score三个数,那么最后就是5+2+12=19

from yolov7_plate.

Ellsionjeep avatar Ellsionjeep commented on May 29, 2024

再次打扰作者一下,我用您给的权重导出onnx后,在yolo_pose_onnx_inference.py的post_process函数中出现The truth value of an array with more than one element is ambiguous. Use a.any() or a.all(),我打印了det_scores[idx].shape 输出是(19,)这一部分维度是否有问题

onnx 推理的代码我还没跑过,19是没问题的,因为 xywh score这就是5个了,然后有两类2个分类得分,4个关键点,每个关键点包括x,y,score三个数,那么最后就是5+2+12=19

我还打印了det_bboxes是(1,4,19)det_scores是(1,19),det_labels是(1,19),kpts是(1,25194,19),就感觉这个onnx推理跑不通

from yolov7_plate.

we0091234 avatar we0091234 commented on May 29, 2024

再次打扰作者一下,我用您给的权重导出onnx后,在yolo_pose_onnx_inference.py的post_process函数中出现The truth value of an array with more than one element is ambiguous. Use a.any() or a.all(),我打印了det_scores[idx].shape 输出是(19,)这一部分维度是否有问题

onnx 推理的代码我还没跑过,19是没问题的,因为 xywh score这就是5个了,然后有两类2个分类得分,4个关键点,每个关键点包括x,y,score三个数,那么最后就是5+2+12=19

我还打印了det_bboxes是(1,4,19)det_scores是(1,19),det_labels是(1,19),kpts是(1,25194,19),就感觉这个onnx推理跑不通

找时间写个onnx推理的代码

from yolov7_plate.

miracletrespasser avatar miracletrespasser commented on May 29, 2024

export onnx模型时出现了这个错误 NYI:Named tensors are not supported with the tracer, 可能是在调用models/export.py 的第80行导致的,nn.Squential不支持命名向量,请问目前有什么好的解决办法吗?

from yolov7_plate.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.