23.11
|
Detection Output layer info. More...
#include <Types.h>
Public Member Functions | |
DetectionOutputLayerInfo () | |
Default Constructor. More... | |
DetectionOutputLayerInfo (int num_classes, bool share_location, DetectionOutputLayerCodeType code_type, int keep_top_k, float nms_threshold, int top_k=-1, int background_label_id=-1, float confidence_threshold=std::numeric_limits< float >::lowest(), bool variance_encoded_in_target=false, float eta=1) | |
Constructor. More... | |
int | num_classes () const |
Get num classes. More... | |
bool | share_location () const |
Get share location. More... | |
DetectionOutputLayerCodeType | code_type () const |
Get detection output code type. More... | |
bool | variance_encoded_in_target () const |
Get if variance encoded in target. More... | |
int | keep_top_k () const |
Get the number of total bounding boxes to be kept per image. More... | |
float | nms_threshold () const |
Get nms threshold. More... | |
float | eta () const |
Get eta. More... | |
int | background_label_id () const |
Get background label ID. More... | |
float | confidence_threshold () const |
Get confidence threshold. More... | |
int | top_k () const |
Get top K. More... | |
int | num_loc_classes () const |
Get number of location classes. More... | |
|
inline |
|
inline |
Constructor.
[in] | num_classes | Number of classes to be predicted. |
[in] | share_location | If true, bounding box are shared among different classes. |
[in] | code_type | Type of coding method for bbox. |
[in] | keep_top_k | Number of total bounding boxes to be kept per image after NMS step. |
[in] | nms_threshold | Threshold to be used in NMS. |
[in] | top_k | (Optional) Number of boxes per image with top confidence scores that are fed into the NMS algorithm. Default set to -1. |
[in] | background_label_id | (Optional) Background label ID. If there is no background class, set it as -1. |
[in] | confidence_threshold | (Optional) Only consider detections whose confidences are larger than a threshold. Default set to -FLT_MAX. |
[in] | variance_encoded_in_target | (Optional) If true, variance is encoded in target. Otherwise we need to adjust the predicted offset accordingly.Default set to false. |
[in] | eta | (Optional) Eta. |
Definition at line 822 of file Types.h.
|
inline |
Get background label ID.
Definition at line 882 of file Types.h.
Referenced by CPPDetectionOutputLayer::configure(), arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get detection output code type.
Definition at line 857 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get confidence threshold.
Definition at line 887 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get eta.
Definition at line 877 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get the number of total bounding boxes to be kept per image.
Definition at line 867 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get nms threshold.
Definition at line 872 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get num classes.
Definition at line 847 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get number of location classes.
Definition at line 897 of file Types.h.
Referenced by CPPDetectionOutputLayer::configure(), arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get share location.
Definition at line 852 of file Types.h.
Referenced by CPPDetectionOutputLayer::configure(), arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get top K.
Definition at line 892 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get if variance encoded in target.
Definition at line 862 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().