CODEWITHSUNDEEP
pythontensorflowobject-detection-apitensorflow-model-garden
Shaaa
Shaaa

Reputation: 23

How can I change the default output, input tensor names in tensorflow object detection API?

Tensorflow object detection API by default has output tensor names num_detections , detection_boxes, detection_scores and detection_classes . How can I change these default names ? Unfortunately an inference platform that I am testing does not accept tensor names with underscores. Thank you ..

Upvotes: 0

Views: 549

Answers (1)

Dieter Maes
Dieter Maes

Reputation: 161

The output tensor names are defined in the "meta_architectures" python files in the object detection project which can be found here. You could try to adjust/convert the naming of the output tensors there for when you are doing the model export for inference but I haven't tried this myself.

Upvotes: 1

Related Questions