Nms

Functions

template<typename Anchor>
std::vector<size_t> nms(const std::vector<Anchor> &boxes, const std::vector<float> &scores, const float thresh, bool includeBoundaries = false, size_t keep_top_k = 0)

std::vector<size_t> multiclass_nms(const std::vector<AnchorLabeled> &boxes, const std::vector<float> &scores, const float iou_threshold = 0.45f, bool includeBoundaries = false, size_t maxNum = 200)

struct Anchor

Subclassed by AnchorLabeled

Public Functions

Anchor() = default

inline Anchor(float _left, float _top, float _right, float _bottom)

inline float getWidth() const

inline float getHeight() const

inline float getXCenter() const

inline float getYCenter() const

Public Members

float left

float top

float right

float bottom

struct AnchorLabeled : public Anchor

Public Functions

AnchorLabeled() = default

inline AnchorLabeled(float _left, float _top, float _right, float _bottom, int _labelID)

Public Members

int labelID = -1