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/how-to-auto-train-image-models.md
+4-3Lines changed: 4 additions & 3 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -533,12 +533,12 @@ In general, deep learning model performance can often improve with more data. Da
533
533
|Object detection, instance segmentation| Training <br><br> Validation & Test |Random crop around bounding boxes, expand, horizontal flip, normalization, resize <br><br><br>Normalization, resize
534
534
|Object detection using yolov5| Training <br><br> Validation & Test |Mosaic, random affine (rotation, translation, scale, shear), horizontal flip <br><br><br> Letterbox resizing|
535
535
536
-
Currently the augmentations defined above are applied by default for an Automated ML job. To facilitate a better control over the augmentations, Automated ML expose flags to turn-off certain train time augmentations. Currently, these flags are only supported forobject detection and instance segmentation tasks. We expose following two flags to turn-off the augmentations.
536
+
Currently the augmentations defined above are applied by default for an Automated ML job. To provide control over augmentations, Automated ML expose flags to turn-off certain augmentations. Currently, these flags are only supported forobject detection and instance segmentation tasks. We expose following two flags to turn-off the augmentations.
537
537
1. **apply_mosaic_for_yolo:** This flag is only specific to Yolo model and turns off the mosaic data augmentation which is applied at the train time.
538
538
2. **apply_automl_train_augmentations:** This flag turns off the augmentation applied at the train time to the object detection and instance segmentation models. For augmentations, please see the details in the table above. Please note that for non-yolo object detection model and instance segmentation model, this flag turns off only first three augmentations i.e. *Random crop around bounding boxes, expand, horizontal flip*. The normalization and resize augmentation is still applied regardless of this flag.
539
-
For Yolo model, this flag turns off the random affine and horizontal flg augmentation.
539
+
For Yolo model, this flag turns off the random affine and horizontal flip augmentation.
540
540
541
-
These two flags are supported via *advanced_settings* under *training_parameters*and can be controlled in the following way.
541
+
These two flags are supported via *advanced_settings* under *training_parameters*and can be controlled in the following way.
In our experiments, we find that these augmentations help the model to generalize better; therefore, when these augmentations are switched off, we recommend the users to combine them with other offline augmentations to get better results.
562
563
## Incremental training (optional)
563
564
564
565
Once the training run is done, you have the option to further train the model by loading the trained model checkpoint. You can either use the same dataset or a different one for incremental training.
0 commit comments