I’ve trained a DetectNet on DIGITS by following some tutorials. Exported the network (Caffe) and wanted to use TensorRT to do the inference with Python.
I had to remove the last layer from my [i]deploy.prototxt[i]. It was a Python layer for clustering. Now I don’t know how to interpret the data that comes out.
I don’t know if I made a mistake when generating the engine, or I just need to do the clustering of my network output, or if I set the input correctly.
I’m sharing my code below so you guys ca take a look:
- Output: http://dontpad.com/e-trt-log
- Output: http://dontpad.com/trt-out
By running this on DIGITS, I got only one bounding box, [952 320 1075 358]. But none of the outputs are even close to this.
How do I fix my current output so I can get the actual bounding boxes? Am I passing the input correctly? Since I have multiple classes, how do I get each bounding box class?
Thank you in advance.