2 instantiations of ImageClassificationTrainer
Microsoft.ML.Vision (2)
VisionCatalog.cs (2)
103new ImageClassificationTrainer(CatalogUtils.GetEnvironment(catalog), options); 130return new ImageClassificationTrainer(CatalogUtils.GetEnvironment(catalog), labelColumnName,
64 references to ImageClassificationTrainer
Microsoft.ML.AutoML (2)
SweepableEstimator\Estimators\Images.cs (1)
45var option = new ImageClassificationTrainer.Options
TrainerExtensions\MultiTrainerExtensions.cs (1)
13using static Microsoft.ML.Vision.ImageClassificationTrainer;
Microsoft.ML.PerformanceTests (4)
ImageClassificationBench.cs (4)
80var options = new ImageClassificationTrainer.Options() 84Arch = ImageClassificationTrainer.Architecture.ResnetV250, 88EarlyStoppingCriteria = new ImageClassificationTrainer.EarlyStopping(minDelta: 0.001f, patience: 20, metric: ImageClassificationTrainer.EarlyStoppingMetric.Loss),
Microsoft.ML.Samples (8)
Dynamic\Trainers\MulticlassClassification\ImageClassification\LearningRateSchedulingCifarResnetTransferLearning.cs (2)
78var options = new ImageClassificationTrainer.Options() 86Arch = ImageClassificationTrainer.Architecture.ResnetV2101,
Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningEarlyStopping.cs (4)
64var options = new ImageClassificationTrainer.Options() 71Arch = ImageClassificationTrainer.Architecture.ResnetV2101, 83EarlyStoppingCriteria = new ImageClassificationTrainer.EarlyStopping( 85metric: ImageClassificationTrainer.EarlyStoppingMetric.Loss),
Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningTrainTestSplit.cs (2)
64var options = new ImageClassificationTrainer.Options() 71Arch = ImageClassificationTrainer.Architecture.ResnetV2101,
Microsoft.ML.Samples.GPU (8)
docs\samples\Microsoft.ML.Samples\Dynamic\Trainers\MulticlassClassification\ImageClassification\LearningRateSchedulingCifarResnetTransferLearning.cs (2)
78var options = new ImageClassificationTrainer.Options() 86Arch = ImageClassificationTrainer.Architecture.ResnetV2101,
docs\samples\Microsoft.ML.Samples\Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningEarlyStopping.cs (4)
64var options = new ImageClassificationTrainer.Options() 71Arch = ImageClassificationTrainer.Architecture.ResnetV2101, 83EarlyStoppingCriteria = new ImageClassificationTrainer.EarlyStopping( 85metric: ImageClassificationTrainer.EarlyStoppingMetric.Loss),
docs\samples\Microsoft.ML.Samples\Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningTrainTestSplit.cs (2)
64var options = new ImageClassificationTrainer.Options() 71Arch = ImageClassificationTrainer.Architecture.ResnetV2101,
Microsoft.ML.TensorFlow.Tests (22)
TensorflowTests.cs (22)
1458internal (string, string, string, bool) getInitialParameters(ImageClassificationTrainer.Architecture arch, string finalImagesFolderName) 1468[InlineData(ImageClassificationTrainer.Architecture.ResnetV2101)] 1469[InlineData(ImageClassificationTrainer.Architecture.MobilenetV2)] 1470[InlineData(ImageClassificationTrainer.Architecture.ResnetV250)] 1471[InlineData(ImageClassificationTrainer.Architecture.InceptionV3)] 1472public void TensorFlowImageClassification(ImageClassificationTrainer.Architecture arch) 1500var options = new ImageClassificationTrainer.Options() 1630workspacePath, isReuse) = getInitialParameters(ImageClassificationTrainer.Architecture.ResnetV2101, _finalImagesFolderName); 1634var options = new ImageClassificationTrainer.Options() 1641Arch = ImageClassificationTrainer.Architecture.ResnetV2101, 1650if (metric.Train.DatasetUsed == ImageClassificationTrainer.ImageClassificationMetrics.Dataset.Train) 1749Assert.True(File.Exists(Path.Combine(Path.GetTempPath(), "MLNET", ImageClassificationTrainer.ModelFileName[options.Arch]))); 1757[InlineData(ImageClassificationTrainer.EarlyStoppingMetric.Accuracy)] 1758[InlineData(ImageClassificationTrainer.EarlyStoppingMetric.Loss)] 1759public void TensorFlowImageClassificationEarlyStopping(ImageClassificationTrainer.EarlyStoppingMetric earlyStoppingMetric) 1787workspacePath, isReuse) = getInitialParameters(ImageClassificationTrainer.Architecture.ResnetV2101, _finalImagesFolderName); 1791var options = new ImageClassificationTrainer.Options() 1798Arch = ImageClassificationTrainer.Architecture.ResnetV2101, 1799EarlyStoppingCriteria = new ImageClassificationTrainer.EarlyStopping(metric: earlyStoppingMetric), 1870var options = new ImageClassificationTrainer.Options() 1877Arch = ImageClassificationTrainer.Architecture.ResnetV2101, 1885var pipeline = _mlContext.MulticlassClassification.Trainers.ImageClassification(options);
Microsoft.ML.Vision (20)
ImageClassificationTrainer.cs (13)
32[assembly: LoadableClass(ImageClassificationTrainer.Summary, typeof(ImageClassificationTrainer), 33typeof(ImageClassificationTrainer.Options), 35ImageClassificationTrainer.UserName, 36ImageClassificationTrainer.LoadName, 37ImageClassificationTrainer.ShortName)] 344/// Options class for <see cref="ImageClassificationTrainer"/>. 497/// Initializes a new instance of <see cref="ImageClassificationTrainer"/> 523/// Initializes a new instance of <see cref="ImageClassificationTrainer"/> 564var logger = Host.Start(nameof(ImageClassificationTrainer)); 1336/// Trains a <see cref="ImageClassificationTrainer"/> using both training and validation data, 1467private readonly ImageClassificationTrainer.ImageProcessor _imageProcessor; 1472_imageProcessor = new ImageClassificationTrainer.ImageProcessor(model._session,
VisionCatalog.cs (7)
88/// Create <see cref="Microsoft.ML.Vision.ImageClassificationTrainer"/> using advanced options, which trains a Deep Neural Network(DNN) to classify images. 91/// <param name="options">An <see cref="ImageClassificationTrainer.Options"/> object specifying advanced 92/// options for <see cref="ImageClassificationTrainer"/>.</param> 100public static ImageClassificationTrainer ImageClassification( 102ImageClassificationTrainer.Options options) => 106/// Create <see cref="Microsoft.ML.Vision.ImageClassificationTrainer"/>, which trains a Deep Neural Network(DNN) to classify images. 121public static ImageClassificationTrainer ImageClassification(