I am using Jetson Nano with Jetpack 4.2
Tensor RT Version : 184.108.40.206
Tensorflow version: 1.13.1 (Nano and Machine on which I export froz)
CUDNN version : 7.3.1
I have downloaded SSD Mobilenet V2 model from https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
I have changed num_classes to 1 since I have only 1 class.
I have commented “batch_norm_trainable: true” this line since I was getting error for this.
Also, made changes related to checkpoints and label.pbtxt.
I have exported graph using script export_inference_graph.py from research/object_detection/ (Not done on Jetson Nano) .
In file config/model_ssd_mobilenet_v2_coco_2018_03_29.py I made following changes:
i) Changed frozen file path (Generated by export_frozen_graph.py)
ii) Change numClasses to 2 since I have only 1 class.
And then run main.py
I got following error:
[TensorRT] ERROR: UFFParser: Validator error: Cast: Unsupported operation _Cast
So, I added following line :
But I got the error,
“[TensorRT] ERROR: UFFParser: Parser error: BoxPredictor_0/Reshape: Reshape: -1 dimension specified more than 1 time”