jetson nano 上 TensorRt 模型(pytorch->onnx->TensorRT)精度下降

Decrease in accuracy of TensorRt model (pytorch->onnx->TensorRT) on jetson nano

我正在 jetson nano 上推断 TensorRt 模型。它是由pytorch通过onnx转换过来的。 在推理时,TensorRt 模型的准确率急剧下降。 它是一个对象检测类型模型。

link 代码: https://github.com/NVIDIA/TensorRT/issues/467

我找到问题了。我不知道为什么,但输入类型会影响模型的准确性。

我已经更换了

img_raw = cv2.imread(image_path, cv2.IMREAD_COLOR)

来自

img_raw = imageio.imread(image_path)

如果我知道背后的原因,我会更新答案。