Skip to content

Commit

Permalink
Update detectionoutput.cpp
Browse files Browse the repository at this point in the history
  • Loading branch information
lzx1413 committed Oct 9, 2017
1 parent 6dab1ea commit 46a5e33
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions src/layer/detectionoutput.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -198,7 +198,7 @@ int DetectionOutput::forward(const std::vector<Mat>& bottom_blobs, std::vector<M
const float* priorbox_ptr = priorbox.row(0);
const float* variance_ptr = priorbox.row(1);

#pragma omp parallel for
#pragma omp parallel for
for (int i = 0; i < num_prior; i++)
{
const float* loc = location_ptr + i * 4;
Expand Down Expand Up @@ -231,7 +231,7 @@ int DetectionOutput::forward(const std::vector<Mat>& bottom_blobs, std::vector<M
all_class_bbox_scores.resize(num_class);

// start from 1 to ignore background class
#pragma omp parallel for
#pragma omp parallel for
for (int i = 1; i < num_class; i++)
{
// filter by confidence_threshold
Expand Down Expand Up @@ -321,5 +321,4 @@ int DetectionOutput::forward(const std::vector<Mat>& bottom_blobs, std::vector<M

return 0;
}

} // namespace ncnn
} // namespace ncnn

0 comments on commit 46a5e33

Please sign in to comment.