专栏算法工具链请问如何去除 Yolov5s 模型最后的 Dequant 层？

请问如何去除 Yolov5s 模型最后的 Dequant 层？

已解决
Shuan2022-08-26
106
1

用户您好，请详细描述您所遇到的问题：

  1. 系统软件版本： XJ3 最新 OE

  2. 问题涉及的技术领域： 算法

  3. 问题描述：希望可以直接拿到 Yolov5s 的 BPU 输出（即使是 INT32/INT8 的），自定义在 CPU 上去做 Dequant 和后处理。

  4. 复现概率：必现 已进行的排查措施、分析及结果：

    1. 硬件问题先排查供电和时钟

    2. 驱动问题先排查外设的供电、复位和时钟

    3. 功能异常，先排查一下是不是运行的代码和demo存在差异，修改不多的情况下直接用对比软件先对比

  5. 提供必要的问题日志： 无

  6. 软件上是否有做自定义修改：无

算法工具链
征程3征程5
评论1
0/1000
  • Horizon_guosheng
    Lv.1

    • 为了更好地适配后处理代码，我们在ONNX模型导出前对Github代码做了如下修改 （代码参见：https://github.com/ultralytics/yolov5/blob/v2.0/models/yolo.py）：

    • 说明: 去除了每个输出分支尾部从4维到5维的reshape（即不将channel从255拆分成3x85），然后将layout从NHWC转换成NCHW再输出。
    2022-11-12
    0
    0