jetson nano 上 TensorRt 模型(pytorch->onnx->TensorRT)精度下降
Decrease in accuracy of TensorRt model (pytorch->onnx->TensorRT) on jetson nano
我正在 jetson nano 上推断 TensorRt 模型。它是由pytorch通过onnx转换过来的。
在推理时,TensorRt 模型的准确率急剧下降。
它是一个对象检测类型模型。
我找到问题了。我不知道为什么,但输入类型会影响模型的准确性。
我已经更换了
img_raw = cv2.imread(image_path, cv2.IMREAD_COLOR)
来自
img_raw = imageio.imread(image_path)
如果我知道背后的原因,我会更新答案。
我正在 jetson nano 上推断 TensorRt 模型。它是由pytorch通过onnx转换过来的。 在推理时,TensorRt 模型的准确率急剧下降。 它是一个对象检测类型模型。
我找到问题了。我不知道为什么,但输入类型会影响模型的准确性。
我已经更换了
img_raw = cv2.imread(image_path, cv2.IMREAD_COLOR)
来自
img_raw = imageio.imread(image_path)
如果我知道背后的原因,我会更新答案。