tutorials icon indicating copy to clipboard operation
tutorials copied to clipboard

Converting from .pth to .onnx failed on 3d-input. Support 3d-conv?

Open Joevaen opened this issue 5 years ago • 1 comments

Converting from .pth to .onnx failed on 3d-input. Support 3d-conv?

Question

Hi, Below picture is the result of a transformation from pth to onnx who are shown in Neutron: image Left is pth's architecture and right is onnx's architecture. You noticed that the first layer size of left input is different from the right. In my exprienment, as for the onnx model, the input data start flowing from the bottom layer(32x1x3x3x3) which is opppsite from the correct situation (flowing from (320x640x3x3x3)). My origin network of .pth includes conv3d and the input data is 5-dim. Can the onnx be converted to onnx on conv3d?

Joevaen avatar Aug 13 '20 01:08 Joevaen

ONNX does support Nd convolution and hence should support Conv3d. As for your PyTorch export to ONNX concerns this might be a question for the PyTorch ONNX export project. Please raise your concern with the PyTorch project. Thanks.

hariharans29 avatar Aug 13 '20 19:08 hariharans29