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

DetectionOutputLayer

Open GraphicsHunter opened this issue 7 years ago • 3 comments

Hi. I tried to implement the detection output layer as you suggested in the previous issue. But when I run my implementation of that layer I always get 200(which is what my keep_top_k value is set to) detections, even when there are no objects on the screen. If I look at the number of detections with Caffe, I see that the numbers are dynamic and depends on if the object is on the screen or not. Do you also have the same "problem" with your implementation?

GraphicsHunter avatar Jan 31 '18 14:01 GraphicsHunter

Top_k doesn't represent the number of detections with caffe and the final output. You need to look at the code again, and remember to verify that the data of every layer is the same as caffe's output.

chenzhi1992 avatar Feb 01 '18 00:02 chenzhi1992

I get success when I run zf-ssd (300x300),but run resnet34-ssd (330x330) was wrong.Detection_output have the same bottoms with caffe.And detection_output's parameters all is right. I suspect the detection_output layer is a problem.

cp625128752 avatar Jul 18 '18 02:07 cp625128752

@chenzhi1992

cp625128752 avatar Jul 18 '18 08:07 cp625128752