search

enpeizhao / CVprojects

computer vision projects | 计算机视觉相关好玩的AI项目（Python、C++、embedded system）
2.15k stars 664 forks source link

关于onnx文件 #30

Closed xingmche closed 6 months ago

xingmche commented 1 year ago

您好，请问这个yolov5s_person.onnx是从哪里得到的啊？我直接从官方下了文件yolov5s.onnx似乎用不了QAQ

yujingli1 commented 6 months ago

[你从他给的下载地址下载吧，这个作者大佬把模型的输出层改了，方便在源码中模型输出调用结果 屏幕截图 2024-04-03 223553 他在源码中这样调用的模型输出结果： 屏幕截图 2024-04-03 223722

yujingli1 commented 6 months ago

作者大大，可能是出于简化和优化代码的想法吧，但是yolov5模型输出层访问起来速度是比较快的，不像yolov8那样修改了模型输出层之后对他的访问会拖慢程序运行时间，如果想用自己的下的官方的模型可以： const float prob = static_cast<const float>(buffers.getHostBuffer(outputTensorName)); outputTensorName写自己模型输出层的名称，然后再自己写模型数据的解析，组成Detection类输入容器

xingmche commented 6 months ago

作者大大，可能是出于简化和优化代码的想法吧，但是yolov5模型输出层访问起来速度是比较快的，不像yolov8那样修改了模型输出层之后对他的访问会拖慢程序运行时间，如果想用自己的下的官方的模型可以： const float prob = static_cast<const float>(buffers.getHostBuffer(outputTensorName)); outputTensorName写自己模型输出层的名称，然后再自己写模型数据的解析，组成Detection类输入容器

多谢老哥解答！

yujingli1 commented 6 months ago

私信了作者大大怎么弄这样的输出层，他说删除原有的decode模块