Share via


ObjectDetectionTrainer.Options Class

Definition

public sealed class ObjectDetectionTrainer.Options : Microsoft.ML.Transforms.TransformInputBase
type ObjectDetectionTrainer.Options = class
    inherit TransformInputBase
Public NotInheritable Class ObjectDetectionTrainer.Options
Inherits TransformInputBase
Inheritance
ObjectDetectionTrainer.Options

Constructors

ObjectDetectionTrainer.Options()

Fields

BoundingBoxColumnName

The Bounding Box column name.

ImageColumnName

The Image column name.

InitLearningRate

Gets or sets the initial learning rate in optimizer.

IOUThreshold

Gets or sets the IOU threshold for removing duplicate bounding boxes.

LabelColumnName

The label column name.

LogEveryNStep

How often to log the loss.

MaxEpoch

Stop training when reaching this number of epochs.

PredictedBoundingBoxColumnName

The Predicted Bounding Box column name.

PredictedLabelColumnName

The label column name.

ScoreColumnName

The Confidence column name.

ScoreThreshold

Gets or sets the confidenct threshold for bounding box category.

Steps

Gets or sets the epoch steps in learning rate scheduler to reduce learning rate.

ValidationSet

The validation set used while training to improve model quality.

WeightDecay

Gets or sets the weight decay in optimizer.

Applies to