Changing infer-dims=3;224;224 to infer-dims=3;224;224;0 produces the error:
Error. 'infer-dims' array length is 4. Should be 3 as [c;h;w] order.
Failed to parse group property
Did you mean input-dims=3;224;224;0
I did that and it didn’t make any difference.
In the mean time I have been trying to follow a different path: train TensoFlow model and convert it to onnx (https://www.tensorflow.org/tutorials/images/transfer_learning). I managed to do it using Resnet50V2 as a base (Resnet50 didn’t work since it had to be converted to onnx with opset 10 which jetson nano TensorRT doesn’t support) and it all worked. I used this config file:
## 0=FP32, 1=INT8, 2=FP16 mode
I am going to give NVIDIA’s tutorial another go and train Resnet model with 50 layers and see if it makes any difference. It does take alot of time though to train on about 7000 images.
I think you can close this ticket since I found a different solution to my problem but I do think that the tutorial needs polishing.