graph.channel_dims这个函数有什么作用？

[394781865]

[xiaoweiChen]

看起来，是在解析上采样时，给反卷积(转置卷积)用的。如果没有在解析onnx模型时保存各层的维度信息，那么将上采样resize或upsample解析成反卷积时，就没办法获取输出通道数信息，而这个输出通道数是要作为卷积参数设置到这个层中间去的。

其他方面就没有看到具体使用了，如果解析onnx模型时，保存了各个层的维度信息，那么graph.channel_dims也可以不用。

[wolfworld6]

看起来，是在解析上采样时，给反卷积(转置卷积)用的。如果没有在解析onnx模型时保存各层的维度信息，那么将上采样resize或upsample解析成反卷积时，就没办法获取输出通道数信息，而这个输出通道数是要作为卷积参数设置到这个层中间去的。

其他方面就没有看到具体使用了，如果解析onnx模型时，保存了各个层的维度信息，那么graph.channel_dims也可以不用。

您好，我在运行时会遇到如下错误，请问是何原因？谢谢
File "yolov5_onnx2caffe/onnx2caffe/_operators.py", line 136, in _convert_sigmoid
graph.channel_dims[output_name] = graph.channel_dims[input_name]
KeyError: '397'

[Hiwyl]

@wolfworld6 你好，我和你遇到了同样的问题，请问您解决了嘛

[swithmn1]

@wolfworld6 你好，我和你遇到同样的问题，请问解决了么

[Robinxin123]

@wolfworld6 ，大佬，解决了吗

[wolfworld6]

@wolfworld6 你好，我和你遇到了同样的问题，请问您解决了嘛

修改onnx的op节点为10，或者修改channel_dims的索引

[wolfworld6]

@wolfworld6 你好，我和你遇到同样的问题，请问解决了么

修改onnx的op节点为10，或者修改channel_dims的索引

[wolfworld6]

@wolfworld6 ，大佬，解决了吗

修改onnx的op节点为10，或者修改channel_dims的索引

[WuJiTe]

@wolfworld6 你好，我和你遇到了同样的问题，请问您解决了嘛

修改onnx的op节点为10，或者修改channel_dims的索引

@wolfworld6 想请教您，channel_dims的索引是遍历graph.inputs获得的，应该怎么修改来解决这个报错？