Skip to content

Commit f1c3382

Browse files
Add validation_iou_threshold and change box_iou_thresh to nms_iou_thresh.
1 parent cc5d23a commit f1c3382

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

articles/machine-learning/reference-automl-images-hyperparameters.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,12 @@ This table summarizes hyperparameters specific to the `yolov5` algorithm.
2525
| Parameter name | Description | Default |
2626
| ------------- |-------------|----|
2727
| `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 |
2829
| `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 |
2930
| `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` |
3031
| `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 |
3132
| `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 |
3334

3435

3536
## Model agnostic hyperparameters
@@ -87,6 +88,7 @@ The following hyperparameters are for object detection and instance segmentation
8788
| Parameter name | Description | Default |
8889
| ------------- |-------------|-----|
8990
| `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 |
9092
| `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 |
9193
| `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 |
9294
| `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

Comments
 (0)