Skip to content

Commit b6959f4

Browse files
authored
Merge pull request #188885 from mispa-ms/mispa/masktool_args
add masktool args
2 parents 7ef21e5 + 426f6f3 commit b6959f4

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

articles/machine-learning/how-to-auto-train-image-models.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -493,8 +493,9 @@ Each of the tasks (and some models) have a set of parameters in the `model_setti
493493
| Task | Parameter name | Default |
494494
|--------- |------------- | --------- |
495495
|Image classification (multi-class and multi-label) | `valid_resize_size`<br>`valid_crop_size` | 256<br>224 |
496-
|Object detection, instance segmentation| `min_size`<br>`max_size`<br>`box_score_thresh`<br>`nms_iou_thresh`<br>`box_detections_per_img` | 600<br>1333<br>0.3<br>0.5<br>100 |
496+
|Object detection | `min_size`<br>`max_size`<br>`box_score_thresh`<br>`nms_iou_thresh`<br>`box_detections_per_img` | 600<br>1333<br>0.3<br>0.5<br>100 |
497497
|Object detection using `yolov5`| `img_size`<br>`model_size`<br>`box_score_thresh`<br>`nms_iou_thresh` | 640<br>medium<br>0.1<br>0.5 |
498+
|Instance segmentation| `min_size`<br>`max_size`<br>`box_score_thresh`<br>`nms_iou_thresh`<br>`box_detections_per_img`<br>`mask_pixel_score_threshold`<br>`max_number_of_polygon_points`<br>`export_as_image`<br>`image_type` | 600<br>1333<br>0.3<br>0.5<br>100<br>0.5<br>100<br>False<br>JPG|
498499

499500
For a detailed description on task specific hyperparameters, please refer to [Hyperparameters for computer vision tasks in automated machine learning](reference-automl-images-hyperparameters.md).
500501

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

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,15 @@ This table summarizes hyperparameters specific to the `yolov5` algorithm.
3232
| `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 |
3333
| `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 |
3434

35+
This table summarizes hyperparameters specific to the `maskrcnn_*` for instance segmentation during inference.
36+
37+
| Parameter name | Description | Default |
38+
| ------------- |-------------|----|
39+
| `mask_pixel_score_threshold` | Score cutoff for considering a pixel as part of the mask of an object. | 0.5 |
40+
| `max_number_of_polygon_points` | Maximum number of (x, y) coordinate pairs in polygon after converting from a mask. | 100 |
41+
| `export_as_image` | Export masks as images. | False |
42+
| `image_type` | Type of image to export mask as (options are jpg, png, bmp). | JPG |
43+
3544
## Model agnostic hyperparameters
3645

3746
The following table describes the hyperparameters that are model agnostic.

0 commit comments

Comments
 (0)