From 3a27cb715fc049624b4e98d9993aabd3694a4bed Mon Sep 17 00:00:00 2001 From: nihuini Date: Wed, 25 Apr 2018 11:13:16 +0800 Subject: [PATCH] skip background class in gathering --- src/layer/detectionoutput.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layer/detectionoutput.cpp b/src/layer/detectionoutput.cpp index 67fe5d4b8..fc12955a7 100644 --- a/src/layer/detectionoutput.cpp +++ b/src/layer/detectionoutput.cpp @@ -239,7 +239,7 @@ int DetectionOutput::forward(const std::vector& bottom_blobs, std::vector bbox_rects; std::vector bbox_scores; - for (int i = 0; i < num_class; i++) + for (int i = 1; i < num_class; i++) { const std::vector& class_bbox_rects = all_class_bbox_rects[i]; const std::vector& class_bbox_scores = all_class_bbox_scores[i];