21.02
|
Classes | |
struct | Box |
Box representation structure. More... | |
struct | Detection |
Detection structure. More... | |
struct | NMSConfig |
Non Maxima Suprresion configuration meta-data. More... | |
Functions | |
bool | compare_detection (const yolov3::Detection &detection, const std::vector< float > &expected) |
Compare a detection object with a vector of float values. More... | |
void | print_detection (std::ostream &os, const std::vector< Detection > &detections) |
Print identified yolo detections. More... | |
std::vector< Detection > | nms (const NMSConfig &config, const std::vector< float > &detected_boxes) |
Perform Non-Maxima Supression on a list of given detections. More... | |
bool compare_detection | ( | const yolov3::Detection & | detection, |
const std::vector< float > & | expected | ||
) |
Compare a detection object with a vector of float values.
detection | [in] Detection object |
expected | [in] Vector of expected float values |
Definition at line 84 of file NMS.cpp.
References Detection::box, Detection::classes, Detection::confidence, Box::xmax, Box::xmin, Box::ymax, and Box::ymin.
Referenced by CheckAccuracy().
std::vector< Detection > nms | ( | const NMSConfig & | config, |
const std::vector< float > & | detected_boxes | ||
) |
Perform Non-Maxima Supression on a list of given detections.
[in] | config | Configuration metadata for NMS |
[in] | detected_boxes | Detected boxes |
Definition at line 113 of file NMS.cpp.
References Detection::classes, NMSConfig::iou_threshold, and NMSConfig::num_classes.