deep-learning - Jetson nano 上 TensorRt 模型（pytorch->onnx->TensorRT）的准确性降低

Question

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

代码链接： https ://github.com/NVIDIA/TensorRT/issues/467

Answer 1

我已经弄清楚了问题所在。我不知道为什么，但输入类型会影响模型的准确性。

我有更换

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

经过

img_raw = imageio.imread(image_path)

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