egbertYeah/mt-yolov6_tensorrt

推理部分输出结果的疑问

Opened this issue · 0 comments

ysj-xuanyuan commented

您好，我在运行代码时发现utils/inference.py中的137行是将context预测的结果进行reshape，但是，我发现reshape前的尺寸是一个一维向量，这似乎和我预想的输出结果有差别。这样直接reshape之后的结果和onnx推理直接得到的结果会有很大的差异吗？