Search code examples
deep-learningconv-neural-networkobject-detectionnvidia-jetson-nano

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


I am inferencing TensorRt model on jetson nano. It is converted from pytorch through onnx. At the time of inference, the accuracy of TensorRt model has decreased drastically. It is an object detection type model.

link for code: https://github.com/NVIDIA/TensorRT/issues/467


Solution

  • I have figure out the problem. I don't know why but type of input effect the accuracy of model.

    I have replace

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

    by

    img_raw = imageio.imread(image_path)

    I will update the answer if I get to know the reason behind it.