Tensorflow-TensorRT
Tensorflow-TensorRT copied to clipboard
Why output_tensors=["Placeholder:0", "concat_9:0", "mul_9:0"]
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.
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.
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.
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.
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 ?
I wonder how you got your yolov3 model file