Hi,
I have export moded to onnx format using the below command.
python3.6 onnx_export.py --model-dir=cat
And when I try to use this onnx model using the command “python3.6 imagenet-console.py --model=/home/shankar/shankar/resnet18.onnx --input_blob=input_0 --output_blob=output_0 --labels=~/datasets/cat/labels.txt ~/datasets/cat/test/02.jpg cat.jpg” , I get the below error.
WARNING: ONNX model has a newer ir_version (0.0.4) than this parser was built against (0.0.3).
While parsing node number 0 [Conv]:
ERROR: ModelImporter.cpp:296 In function importModel:
[5] Assertion failed: tensors.count(input_name)
[TRT] failed to parse ONNX model ‘/home/shankar/shankar/resnet18.onnx’
[TRT] device GPU, failed to load /home/shankar/shankar/resnet18.onnx
[TRT] failed to load /home/shankar/shankar/resnet18.onnx
[TRT] imageNet – failed to initialize.
Please suggest fix for this.
I am using the latest jetson nano image r32.3.1 (released on dec 10th 2019).
Python: 3.6
PyTorch v1.3.0
Torchvision v0.5.0