Verbose |
True |
True, False |
Prints/logs progress and status messages. |
Model |
'Medium' |
Small, Medium, Large, XSmall, XLarge, custom path |
Model size/name/path selector. |
Variant |
'B' |
A, B, C, custom |
Model variant. |
Channels |
128 |
Any valid value |
Neural feature channel count. |
Inference |
'HBB' |
classify, hbb, hbbv2, segment, pose, obb, obbv2 |
Neural task type such as classify, detect, segment, pose, OBB, or HBB. |
GPU |
True |
True, False |
Uses GPU acceleration where available. |
Layer_Start |
None |
Any valid value |
First layer used for training/inference where supported. |
Layers |
None |
Any valid value |
Layer list/selection. |
FPN_Layers |
None |
Any valid value |
Feature-pyramid layers. |
Data |
False |
Any valid value |
Dataset path or dataset configuration. |
Val_Freq |
1 |
Any valid value |
Validation frequency. |
Val_Subset |
0 |
Any valid value |
Validation subset size/count. |
Image_Size |
False |
Any valid value |
Input image size. |
Learning_Rate |
None |
Any valid value |
Training learning rate. |
Epoch |
20 |
Any valid value |
Number of training epochs. |
Batch_Size |
2 |
Any valid value |
Training/inference batch size. |
Worker |
4 |
Any valid value |
Number of data-loader workers. |
Patience |
None |
Any valid value |
Early-stop patience. |
Step |
None |
Any valid value |
Learning-rate step schedule. |
Gamma |
0.1 |
Any valid value |
Learning-rate decay factor. |
Momentum |
0.9 |
Any valid value |
Optimizer momentum. |
Weight_Decay |
None |
Any valid value |
Weight decay regularization. |
Callback |
False |
Any valid value |
Optional progress callback. |
Threshold |
0.5 |
Any valid value |
Confidence threshold. |
NMS |
0.5 |
Any valid value |
Non-maximum suppression overlap threshold. |
NMS_Batch |
False |
Any valid value |
Batch NMS setting. |
Test_Threshold |
0.3 |
Any valid value |
Threshold used during testing. |
Margin |
10 |
Any valid value |
Extra region margin. |
Max_Age |
30 |
Any valid value |
Maximum track age before dropping a track. |
Pretrained |
True |
True, False |
Uses pretrained weights where available. |
Export_Opset |
20 |
Any valid value |
ONNX export opset version. |
Mask_Quality |
False |
Any valid value |
Mask quality setting. |
Mask_Layers |
4 |
Any valid value |
Number of mask layers. |
Detections |
300 |
Any valid value |
Maximum detections to keep. |
TopK |
1000 |
Any valid value |
Top-K candidates to keep. |
Pre_Top |
2000 |
Any valid value |
Candidates kept before filtering. |
Post_Top |
1000 |
Any valid value |
Candidates kept after filtering. |
Text_Region |
False |
Any valid value |
Draws/shows region text. |
Text_Class |
True |
Any valid value |
Draws/shows class text. |
Text_Score |
False |
Any valid value |
Draws/shows score text. |
Text_ID |
False |
Any valid value |
Draws/shows track ID text. |
Anchor_Scale |
None |
Any valid value |
Anchor scale list/setting. |
Anchor_Ratio |
None |
Any valid value |
Anchor ratio list/setting. |