You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: articles/machine-learning/reference-automl-images-hyperparameters.md
+3-1Lines changed: 3 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -25,11 +25,12 @@ This table summarizes hyperparameters specific to the `yolov5` algorithm.
25
25
| Parameter name | Description | Default |
26
26
| ------------- |-------------|----|
27
27
|`validation_metric_type`| Metric computation method to use for validation metrics. <br> Must be `none`, `coco`, `voc`, or `coco_voc`. |`voc`|
28
+
|`validation_iou_threshold`| IOU threshold for box matching when computing validation metrics. <br>Must be a float in the range [0.1, 1]. | 0.5 |
28
29
|`img_size`| Image size for train and validation. <br> Must be a positive integer. <br> <br> *Note: training run may get into CUDA OOM if the size is too big*. | 640 |
29
30
|`model_size`| Model size. <br> Must be `small`, `medium`, `large`, or `xlarge`. <br><br> *Note: training run may get into CUDA OOM if the model size is too big*. |`medium`|
30
31
|`multi_scale`| Enable multi-scale image by varying image size by +/- 50% <br> Must be 0 or 1. <br> <br> *Note: training run may get into CUDA OOM if no sufficient GPU memory*. | 0 |
31
32
|`box_score_thresh`| During inference, only return proposals with a score greater than `box_score_thresh`. The score is the multiplication of the objectness score and classification probability. <br> Must be a float in the range [0, 1]. | 0.1 |
32
-
|`box_iou_thresh`| IoU threshold used during inference in non-maximum suppression post processing. <br> Must be a float in the range [0, 1]. | 0.5 |
33
+
|`nms_iou_thresh`| IoU threshold used during inference in non-maximum suppression post processing. <br> Must be a float in the range [0, 1]. | 0.5 |
33
34
34
35
35
36
## Model agnostic hyperparameters
@@ -87,6 +88,7 @@ The following hyperparameters are for object detection and instance segmentation
87
88
| Parameter name | Description | Default |
88
89
| ------------- |-------------|-----|
89
90
|`validation_metric_type`| Metric computation method to use for validation metrics. <br> Must be `none`, `coco`, `voc`, or `coco_voc`. |`voc`|
91
+
|`validation_iou_threshold`| IOU threshold for box matching when computing validation metrics. <br>Must be a float in the range [0.1, 1]. | 0.5 |
90
92
|`min_size`| Minimum size of the image to be rescaled before feeding it to the backbone. <br> Must be a positive integer. <br> <br> *Note: training run may get into CUDA OOM if the size is too big*.| 600 |
91
93
|`max_size`| Maximum size of the image to be rescaled before feeding it to the backbone. <br> Must be a positive integer.<br> <br> *Note: training run may get into CUDA OOM if the size is too big*. | 1333 |
92
94
|`box_score_thresh`| During inference, only return proposals with a classification score greater than `box_score_thresh`. <br> Must be a float in the range [0, 1].| 0.3 |
0 commit comments