Tensorflow-TensorRT icon indicating copy to clipboard operation
Tensorflow-TensorRT copied to clipboard

Why output_tensors=["Placeholder:0", "concat_9:0", "mul_9:0"]

Open float123 opened this issue 5 years ago • 5 comments

I wan to know why output_tensors=["Placeholder:0", "concat_9:0", "mul_9:0"], these three nodes are not at the end of the model. And what does ":0" mean? Please advise,thank you. image

float123 avatar Jun 19 '19 09:06 float123

Why not at the end of model

Because after the network output, we still have further operation, such as loss calculation, etc.

what :0 is

It is tensorflow convention when we wanna call a value from a tensor by tensor name, we need to add :0 in the end of it.

ardianumam avatar Jun 19 '19 09:06 ardianumam

Thank you very much for your reply,I am currently researching yolov3 to TensorRT,your project has helped me a lot. There is still a question I would like to ask,why do we choose output_tensor to be "concat_9" and "mul_9", or what are their return values? the model is very troubled by me.

float123 avatar Jun 20 '19 02:06 float123

Those only tensor names. Everyone can name the tensor any name he wants. Those output in object detector correspond to the outputs of classification and bonding box predictions.

ardianumam avatar Jul 07 '19 03:07 ardianumam

Hey, I am confused. Where did you get this model architecture from? Did you fork YunYang's model. More specifically, why is NMS part of your model ?

Pidem avatar Aug 08 '19 21:08 Pidem

I wonder how you got your yolov3 model file

DongXinru avatar Aug 30 '19 07:08 DongXinru