Difference of the shape between anchors which runs with original model in Pytorch and Onnx.

murdockhou / Yet-Another-EfficientDet-Pytorch-Convert-ONNX-TVM

GNU Lesser General Public License v3.0

59 stars 17 forks source link

Difference of the shape between anchors which runs with original model in Pytorch and Onnx. #8

Open hieubz opened 3 years ago

hieubz commented 3 years ago

Hi @murdockhou I ran with both original model of author and the converted version to Onnx. But I found that the output anchors of original model was torch.Size([1, 196416, 4]), but the output anchors of onnx model was (1, 191997, 4).

I don't know why was that? Would you please explain me about that? I will really appreciate that.

Thank you.