ImageModelSettingsObjectDetection 類別
AutoML 影像物件偵測工作的模型設定。
- 繼承
-
azure.ai.ml.entities._job.automl.image.image_model_settings.ImageModelDistributionSettingsImageModelSettingsObjectDetection
建構函式
ImageModelSettingsObjectDetection(*, advanced_settings: str | None = None, ams_gradient: bool | None = None, beta1: float | None = None, beta2: float | None = None, checkpoint_frequency: int | None = None, checkpoint_run_id: str | None = None, distributed: bool | None = None, early_stopping: bool | None = None, early_stopping_delay: int | None = None, early_stopping_patience: int | None = None, enable_onnx_normalization: bool | None = None, evaluation_frequency: int | None = None, gradient_accumulation_step: int | None = None, layers_to_freeze: int | None = None, learning_rate: float | None = None, learning_rate_scheduler: LearningRateScheduler | None = None, model_name: str | None = None, momentum: float | None = None, nesterov: bool | None = None, number_of_epochs: int | None = None, number_of_workers: int | None = None, optimizer: StochasticOptimizer | None = None, random_seed: int | None = None, step_lr_gamma: float | None = None, step_lr_step_size: int | None = None, training_batch_size: int | None = None, validation_batch_size: int | None = None, warmup_cosine_lr_cycles: float | None = None, warmup_cosine_lr_warmup_epochs: int | None = None, weight_decay: float | None = None, box_detections_per_image: int | None = None, box_score_threshold: float | None = None, image_size: int | None = None, max_size: int | None = None, min_size: int | None = None, model_size: ModelSize | None = None, multi_scale: bool | None = None, nms_iou_threshold: float | None = None, tile_grid_size: str | None = None, tile_overlap_ratio: float | None = None, tile_predictions_nms_threshold: float | None = None, validation_iou_threshold: float | None = None, validation_metric_type: ValidationMetricType | None = None, log_training_metrics: LogTrainingMetrics | None = None, log_validation_loss: LogValidationLoss | None = None, **kwargs)
參數
- gradient_accumulation_step
- int
漸層累積表示在不更新模型權數的情況下執行已設定的 「GradAccumulationStep」 步驟,同時累積這些步驟的漸層,然後使用累積的漸層來計算權數更新。 必須是正整數。
- layers_to_freeze
- int
要凍結模型的圖層數目。 必須是正整數。 例如,將 2 當做 'seresnext' 的值傳遞表示凍結 layer0 和 layer1。 如需支援之模型的完整清單,以及圖層凍結的詳細資料,請參閱: https://docs.microsoft.com/en-us/azure/machine-learning/how-to-auto-train-image-models 。
- learning_rate_scheduler
- str 或 LearningRateScheduler
學習率排程器的類型。 必須是 'warmup_cosine' 或 'step'。 可能的值包括:「None」、「WarmupCosine」、「Step」。
- model_name
- str
要用於定型的模型名稱。 如需可用模型的詳細資訊,請流覽官方檔: https://docs.microsoft.com/en-us/azure/machine-learning/how-to-auto-train-image-models 。
- max_size
- int
將影像饋送至骨幹之前要重新調整的影像大小上限。 必須是正整數。 注意:如果大小太大,則定型執行可能會進入 CUDA OOM. 注意:'yolov5' 演算法不支援此設定。
- min_size
- int
將影像饋送至骨幹之前要重新調整的影像大小下限。 必須是正整數。 注意:如果大小太大,則定型執行可能會進入 CUDA OOM. 注意:'yolov5' 演算法不支援此設定。
模型大小。 必須是 'small'、'medium'、'large'。 注意:如果模型大小太大,則定型執行可能會進入 CUDA OOM. 注意:只有 'yolov5' 演算法才支援此設定。 可能的值為:「None」、「Small」、「Medium」、「Large」、「ExtraLarge」。
- multi_scale
- bool
依 +/- 50% 的不同影像大小啟用多縮放映射。 注意:如果沒有足夠的 GPU 記憶體,則定型執行可能會進入 CUDA OOM。 注意:只有 'yolov5' 演算法才支援此設定。
- tile_grid_size
- str
要用於每個並排影像的格線大小。 注意:TileGridSize 不得為 None,才能啟用小型物件偵測邏輯。 包含兩個整數的字串,其格式為#。 注意:'yolov5' 演算法不支援此設定。
- tile_predictions_nms_threshold
- float
從圖格和影像合併預測時用來執行 NMS 的 IOU 閾值。 用於驗證/推斷。 必須是範圍 [0, 1] 中的 float。 注意:'yolov5' 演算法不支援此設定。
- validation_metric_type
- str 或 ValidationMetricType
要用於驗證計量的計量計算方法。 可能的值包括:「None」、「Coco」、「Voc」、「CocoVoc」。
- log_training_metrics
- str 或 <xref:azure.mgmt.machinelearningservices.models.LogTrainingMetrics>
指出是否要記錄定型計量
- log_validation_loss
- str 或 <xref:azure.mgmt.machinelearningservices.models.LogValidationLoss>
指出是否要記錄驗證遺失