专栏算法工具链关于分割模型导出onnx问题

关于分割模型导出onnx问题

已解决
Sunseeker2022-11-25
48
1

使用export_onnx.py时,默认导出的onnx模型的输出维度为 nchw,如何修改才能使得onnx的输出维度变成 nhwc呢?

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2

    您好,您可以在模型尾部自行插入transpose节点进行尝试,或者在后处理中进行维度变换

    2022-11-28
    0
    0