|
using Microsoft.ML.SearchSpace;
using OptionAttribute = Microsoft.ML.SearchSpace.NestOptionAttribute;
using ColorsOrder = Microsoft.ML.Transforms.Image.ImagePixelExtractingEstimator.ColorsOrder;
using ColorBits = Microsoft.ML.Transforms.Image.ImagePixelExtractingEstimator.ColorBits;
using ResizingKind = Microsoft.ML.Transforms.Image.ImageResizingEstimator.ResizingKind;
using Anchor = Microsoft.ML.Transforms.Image.ImageResizingEstimator.Anchor;
using BertArchitecture = Microsoft.ML.TorchSharp.NasBert.BertArchitecture;
using static Microsoft.ML.Vision.ImageClassificationTrainer.Architecture;
using DataKind = Microsoft.ML.Data.DataKind;
using CaseMode = Microsoft.ML.Transforms.Text.TextNormalizingEstimator.CaseMode;
#nullable enable
namespace Microsoft.ML.AutoML.CodeGen
{
public class DnnFeaturizerImageOption
{
public string? OutputColumnName {get; set;}
public string? InputColumnName {get; set;}
[Choice(new object[]{ "alexnet","resnet_101","resnet_18","resnet_50" }, "resnet_18")]
[System.Diagnostics.CodeAnalysis.SuppressMessage("Declaration", "MSML_NoInstanceInitializers:No initializers on instance fields or properties")]
public string ModelFactory {get; set;} = "resnet_18";
}
}
|