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

439 views Asked by At

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 ..

1

There are 1 answers

0
Dieter Maes On BEST ANSWER

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.