It looks like your model includes a non-supported layer: Cast.
It may be added by TensorFlow automatically as an utility operation.
It’s recommended to check how to remove this operation in TensorFlow when retraining the model.
To remove it directly from graphsurgeon may lead to some unexpected issues.
I solved the Problem with the -1 dimension, by exporting my inference Graph with a fixed Batch size of 1 [1,None,None,3].
but then I got the Error
ERROR: UFFParser: Validator error: FeatureExtractor/MobilenetV2/layer_19_2_Conv2d_4_3x3_s2_256/BatchNorm/FusedBatchNormV3: Unsupported operation _FusedBatchNormV3
I am having this exact same issue training a new Model in Tensorflow based on the checkpoint from model ssd_inception_v2_coco_2017_11_17 in order to use it with TensorRT.
My goal is to create an Object Detection model from custom COCO dataset for use with TensorRT.
I am getting the Cast error and then the other error when I remove Cast operations.
How do I remove CAST operations from the training of a new model with python3 object_detection/model_main.py?
Along with anything else to make it TensorRT compatible.
Also what is the best newest way to make an object detection model that works with TensorRT?