Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
1.0 KiB
33 lines
1.0 KiB
// This file is part of OpenCV project. |
|
// It is subject to the license terms in the LICENSE file found in the top-level directory |
|
// of this distribution and at http://opencv.org/license.html. |
|
// |
|
// Copyright (C) 2017, Intel Corporation, all rights reserved. |
|
// Third party copyrights are property of their respective owners. |
|
|
|
#include "precomp.hpp" |
|
#include "nms.inl.hpp" |
|
|
|
namespace cv |
|
{ |
|
namespace dnn |
|
{ |
|
CV__DNN_EXPERIMENTAL_NS_BEGIN |
|
|
|
static inline float rectOverlap(const Rect& a, const Rect& b) |
|
{ |
|
return 1.f - static_cast<float>(jaccardDistance(a, b)); |
|
} |
|
|
|
void NMSBoxes(const std::vector<Rect>& bboxes, const std::vector<float>& scores, |
|
const float score_threshold, const float nms_threshold, |
|
std::vector<int>& indices, const float eta, const int top_k) |
|
{ |
|
CV_Assert(bboxes.size() == scores.size(), score_threshold >= 0, |
|
nms_threshold >= 0, eta > 0); |
|
NMSFast_(bboxes, scores, score_threshold, nms_threshold, eta, top_k, indices, rectOverlap); |
|
} |
|
|
|
CV__DNN_EXPERIMENTAL_NS_END |
|
}// dnn |
|
}// cv
|
|
|