Skip to content

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