1 write to Options
Microsoft.ML.TorchSharp (1)
AutoFormerV2\ObjectDetectionTrainer.cs (1)
526
Options
= options;
32 references to Options
Microsoft.ML.TorchSharp (32)
AutoFormerV2\ObjectDetectionTrainer.cs (32)
528
PredictedLabelColumnName = new SchemaShape.Column(
Options
.PredictedLabelColumnName, SchemaShape.Column.VectorKind.Vector, NumberDataViewType.UInt32, false);
529
PredictedBoundingBoxColumn = new SchemaShape.Column(
Options
.PredictedBoundingBoxColumnName, SchemaShape.Column.VectorKind.Vector, NumberDataViewType.Single, false);
530
ConfidenceColumn = new SchemaShape.Column(
Options
.ScoreColumnName, SchemaShape.Column.VectorKind.Vector, NumberDataViewType.Single, false);
562
outColumns[
Options
.PredictedLabelColumnName] = new SchemaShape.Column(
Options
.PredictedLabelColumnName, SchemaShape.Column.VectorKind.VariableVector,
565
outColumns[
Options
.PredictedBoundingBoxColumnName] = new SchemaShape.Column(
Options
.PredictedBoundingBoxColumnName, SchemaShape.Column.VectorKind.VariableVector,
568
outColumns[
Options
.ScoreColumnName] = new SchemaShape.Column(
Options
.ScoreColumnName, SchemaShape.Column.VectorKind.VariableVector,
576
if (!inputSchema.TryFindColumn(
Options
.ImageColumnName, out var imageCol))
577
throw Host.ExceptSchemaMismatch(nameof(inputSchema), "Image",
Options
.ImageColumnName);
579
throw Host.ExceptSchemaMismatch(nameof(inputSchema), "Image",
Options
.ImageColumnName,
613
ctx.SaveNonEmptyString(
Options
.LabelColumnName);
614
ctx.SaveNonEmptyString(
Options
.PredictedLabelColumnName);
615
ctx.SaveNonEmptyString(
Options
.BoundingBoxColumnName);
616
ctx.SaveNonEmptyString(
Options
.PredictedBoundingBoxColumnName);
617
ctx.SaveNonEmptyString(
Options
.ImageColumnName);
618
ctx.SaveNonEmptyString(
Options
.ScoreColumnName);
620
ctx.Writer.Write(
Options
.NumberOfClasses);
622
ctx.Writer.Write(
Options
.ScoreThreshold);
623
ctx.Writer.Write(
Options
.IOUThreshold);
736
if (inputSchema.TryGetColumnIndex(parent.
Options
.ImageColumnName, out var col))
767
info[0] = new DataViewSchema.DetachedColumn(_parent.
Options
.PredictedLabelColumnName, new VectorDataViewType(new KeyDataViewType(typeof(uint), _parent.
Options
.NumberOfClasses)), labelBuilder.ToAnnotations());
769
info[1] = new DataViewSchema.DetachedColumn(_parent.
Options
.ScoreColumnName, new VectorDataViewType(NumberDataViewType.Single), meta.ToAnnotations());
771
info[2] = new DataViewSchema.DetachedColumn(_parent.
Options
.PredictedBoundingBoxColumnName, new VectorDataViewType(NumberDataViewType.Single));
808
getImage = input.GetGetter<MLImage>(input.Schema[_parent.
Options
.ImageColumnName]);
832
getImage = input.GetGetter<MLImage>(input.Schema[_parent.
Options
.ImageColumnName]);
856
getImage = input.GetGetter<MLImage>(input.Schema[_parent.
Options
.ImageColumnName]);
881
TensorCacher outputCacher = new TensorCacher(_parent.
Options
.NumberOfClasses);
973
ImageUtils.Postprocess(imageTensor, pred, score, box, out outputCache.PredictedLabelsBuffer, out outputCache.ScoresBuffer, out outputCache.BoxBuffer, _parent.
Options
.ScoreThreshold, _parent.
Options
.IOUThreshold);