Снижение точности модели TensorRt (pytorch-> onnx-> TensorRT) на jetson nano

Question

Я делаю вывод о модели TensorRt на jetson nano. Конвертируется из 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)

Я уточню ответ, если узнаю причину этого.