Browse Source

skip background class in gathering

tags/20180427
nihuini 8 years ago
parent
commit
3a27cb715f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/layer/detectionoutput.cpp

+ 1
- 1
src/layer/detectionoutput.cpp View File

@@ -239,7 +239,7 @@ int DetectionOutput::forward(const std::vector<Mat>& bottom_blobs, std::vector<M
std::vector<BBoxRect> bbox_rects;
std::vector<float> bbox_scores;

for (int i = 0; i < num_class; i++)
for (int i = 1; i < num_class; i++)
{
const std::vector<BBoxRect>& class_bbox_rects = all_class_bbox_rects[i];
const std::vector<float>& class_bbox_scores = all_class_bbox_scores[i];


Loading…
Cancel
Save