how to get layer output

Ghustwb / MobileNet-SSD-TensorRT

Accelerate mobileNet-ssd with tensorRT

188 stars 91 forks source link

how to get layer output #11

Closed ghost closed 5 years ago

ghost commented 5 years ago

thanks for your work and how to get layer output for debug with tensorrt?

Ghustwb commented 5 years ago

Like this, mbox_conf_softmax,you can print it const char OUTPUT1 = "mbox_conf_softmax"; DimsCHW dims1 = tensorNet.getTensorDims(OUTPUT1); float output1 = allocateMemory( dims1 , (char*)"output1");

ghost commented 5 years ago

thx，I had solve this problem like you