Model - Hyperparameter Support
Object Detection - Hyperparameters
Support Grid
Model | MAX_BBOX_PER_IMG | MAX_BBOX_PER_CLS | NMS_PRE | MIN_BBOX_SIZE | IOU | SCORE_THRESHOLD | CONFIDENCE_THRESHOLD |
---|---|---|---|---|---|---|---|
YOLOV8 | YES | YES | YES | NO | YES | YES | YES |
RTMDet | Yes | Yes | Yes | Yes | Yes | Yes | Yes |
YOLOX | NO | Yes | Yes | NO | Yes | Yes | YES |
Hyperparameter Description
Hyperparameter | Definition | Default Value |
---|---|---|
MAX_BOX_PER_IMG | Maximum number of bounding boxes to generate per image | 100 |
MAX_BOX_PER_CLS | Maximum Number of Bounding Boxes to generate per class | 100 |
NMS_PRE | The number of bounding boxes before Non Max Supression | 1000 |
IOU | The IOU Threshold for Non Max Suppression | 0.65 |
SCORE_THRESHOLD | The Threshod used to filter out boxes using class score | 0.03 |
CONFIDENCE_THRESHOLD | The Threshold used to filter out boxes using confidence score | 0.005 |