21.08
|
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 933 of file Types.h.
|
inline |
Get background label ID.
Definition at line 985 of file Types.h.
Referenced by CPPDetectionOutputLayer::configure(), arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get detection output code type.
Definition at line 960 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get confidence threshold.
Definition at line 990 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get eta.
Definition at line 980 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 970 of file Types.h.
Referenced by DetectionOutputLayerNode::compute_output_descriptor(), CPPDetectionOutputLayer::configure(), arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get nms threshold.
Definition at line 975 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get num classes.
Definition at line 950 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get number of location classes.
Definition at line 1000 of file Types.h.
Referenced by CPPDetectionOutputLayer::configure(), arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get share location.
Definition at line 955 of file Types.h.
Referenced by CPPDetectionOutputLayer::configure(), arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get top K.
Definition at line 995 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().
|
inline |
Get if variance encoded in target.
Definition at line 965 of file Types.h.
Referenced by arm_compute::operator<<(), and CPPDetectionOutputLayer::run().