search

WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.99k stars 518 forks source link

How to inference yolor onnx file? #249

Open ninenox-dev opened 2 years ago

ninenox-dev commented 2 years ago

I try with code inference onnx of yolox,It's not working.

line 121, in _postprocess outputs[..., :2] = (outputs[..., :2] + grids) * expanded_strides ValueError: operands could not be broadcast together with shapes (1,12495,2) (1,4116,2)