Closed kHarshit closed 2 years ago
Have you followed the following steps
I saw that you added a maximum pooling with a convolutional kernel size of 3 steps and a padding of 1 after ONNX. Have you performed any other subsequent processing?
Sorry, I haven't put this issue into practice the other day. I also encountered the same problem earlier. I think a certain process was lost during export.
At first, I used the same method as you to export, but I did not infer the correct result. At this point, I believe there may be an error in the flow of the diagram.
Later, I used torch.onnx.export
to export the model, but there were still errors.
I've converted FIDTM model to onnx using following logic, but the output from onnx is wrong.
But after loading this onnx model, the output is wrong.
In fact, the onnx model gives almost the same values for every input image. Is it happening due to
if ... else
blocks in the model? I'm not sure if model is getting converted correctly!