2 instantiations of ImageClassificationTrainer
Microsoft.ML.Vision (2)
VisionCatalog.cs (2)
103
new
ImageClassificationTrainer
(CatalogUtils.GetEnvironment(catalog), options);
130
return new
ImageClassificationTrainer
(CatalogUtils.GetEnvironment(catalog), labelColumnName,
93 references to ImageClassificationTrainer
Microsoft.ML.AutoML (31)
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ApplyOnnxModelOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ConcatOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ConvertTypeOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\DnnFeaturizerImageOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ExtractPixelsOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\FastForestOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\FastTreeOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\FeaturizeTextOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ImageClassificationOption.generated.cs (2)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
33
public Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture Arch {get; set;} = ResnetV250;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\LbfgsOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\LgbmOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\LoadImageOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\MapKeyToValueOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\MapValueToKeyOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\MatrixFactorizationOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\NaiveOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\NamedEntityRecognitionOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\NormalizeMinMaxOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\NormalizeTextOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ObjectDetectionOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\OneHotOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\QuestionAnsweringOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ReplaceMissingValueOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\ResizeImageOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\SdcaOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\SentenceSimilarityOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\SsaOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
Generated\Basic.CompilerLog.Util\Basic.CompilerLog.Util.Impl.BasicGeneratedFilesAnalyzerReference\TextClassificationOption.generated.cs (1)
9
using static Microsoft.ML.Vision.
ImageClassificationTrainer
.Architecture;
SweepableEstimator\Estimators\Images.cs (1)
45
var option = new
ImageClassificationTrainer
.Options
TrainerExtensions\MultiTrainerExtensions.cs (1)
13
using static Microsoft.ML.Vision.
ImageClassificationTrainer
;
Microsoft.ML.PerformanceTests (4)
ImageClassificationBench.cs (4)
80
var options = new
ImageClassificationTrainer
.Options()
84
Arch =
ImageClassificationTrainer
.Architecture.ResnetV250,
88
EarlyStoppingCriteria = new
ImageClassificationTrainer
.EarlyStopping(minDelta: 0.001f, patience: 20, metric:
ImageClassificationTrainer
.EarlyStoppingMetric.Loss),
Microsoft.ML.Samples (8)
Dynamic\Trainers\MulticlassClassification\ImageClassification\LearningRateSchedulingCifarResnetTransferLearning.cs (2)
78
var options = new
ImageClassificationTrainer
.Options()
86
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningEarlyStopping.cs (4)
64
var options = new
ImageClassificationTrainer
.Options()
71
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
83
EarlyStoppingCriteria = new
ImageClassificationTrainer
.EarlyStopping(
85
metric:
ImageClassificationTrainer
.EarlyStoppingMetric.Loss),
Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningTrainTestSplit.cs (2)
64
var options = new
ImageClassificationTrainer
.Options()
71
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
Microsoft.ML.Samples.GPU (8)
parent\6a273e5f1a281752\LearningRateSchedulingCifarResnetTransferLearning.cs (2)
78
var options = new
ImageClassificationTrainer
.Options()
86
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
parent\Microsoft.ML.Samples\Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningEarlyStopping.cs (4)
64
var options = new
ImageClassificationTrainer
.Options()
71
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
83
EarlyStoppingCriteria = new
ImageClassificationTrainer
.EarlyStopping(
85
metric:
ImageClassificationTrainer
.EarlyStoppingMetric.Loss),
parent\Microsoft.ML.Samples\Dynamic\Trainers\MulticlassClassification\ImageClassification\ResnetV2101TransferLearningTrainTestSplit.cs (2)
64
var options = new
ImageClassificationTrainer
.Options()
71
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
Microsoft.ML.TensorFlow.Tests (22)
TensorflowTests.cs (22)
1450
internal (string, string, string, bool) getInitialParameters(
ImageClassificationTrainer
.Architecture arch, string finalImagesFolderName)
1460
[InlineData(
ImageClassificationTrainer
.Architecture.ResnetV2101)]
1461
[InlineData(
ImageClassificationTrainer
.Architecture.MobilenetV2)]
1462
[InlineData(
ImageClassificationTrainer
.Architecture.ResnetV250)]
1463
[InlineData(
ImageClassificationTrainer
.Architecture.InceptionV3)]
1464
public void TensorFlowImageClassification(
ImageClassificationTrainer
.Architecture arch)
1492
var options = new
ImageClassificationTrainer
.Options()
1622
workspacePath, isReuse) = getInitialParameters(
ImageClassificationTrainer
.Architecture.ResnetV2101, _finalImagesFolderName);
1626
var options = new
ImageClassificationTrainer
.Options()
1633
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
1642
if (metric.Train.DatasetUsed ==
ImageClassificationTrainer
.ImageClassificationMetrics.Dataset.Train)
1741
Assert.True(File.Exists(Path.Combine(Path.GetTempPath(), "MLNET",
ImageClassificationTrainer
.ModelFileName[options.Arch])));
1749
[InlineData(
ImageClassificationTrainer
.EarlyStoppingMetric.Accuracy)]
1750
[InlineData(
ImageClassificationTrainer
.EarlyStoppingMetric.Loss)]
1751
public void TensorFlowImageClassificationEarlyStopping(
ImageClassificationTrainer
.EarlyStoppingMetric earlyStoppingMetric)
1779
workspacePath, isReuse) = getInitialParameters(
ImageClassificationTrainer
.Architecture.ResnetV2101, _finalImagesFolderName);
1783
var options = new
ImageClassificationTrainer
.Options()
1790
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
1791
EarlyStoppingCriteria = new
ImageClassificationTrainer
.EarlyStopping(metric: earlyStoppingMetric),
1862
var options = new
ImageClassificationTrainer
.Options()
1869
Arch =
ImageClassificationTrainer
.Architecture.ResnetV2101,
1877
var
pipeline = _mlContext.MulticlassClassification.Trainers.ImageClassification(options);
Microsoft.ML.Vision (20)
ImageClassificationTrainer.cs (13)
32
[assembly: LoadableClass(
ImageClassificationTrainer
.Summary, typeof(
ImageClassificationTrainer
),
33
typeof(
ImageClassificationTrainer
.Options),
35
ImageClassificationTrainer
.UserName,
36
ImageClassificationTrainer
.LoadName,
37
ImageClassificationTrainer
.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
"/>
564
var logger = Host.Start(nameof(
ImageClassificationTrainer
));
1320
/// Trains a <see cref="
ImageClassificationTrainer
"/> using both training and validation data,
1451
private readonly
ImageClassificationTrainer
.ImageProcessor _imageProcessor;
1456
_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>
100
public static
ImageClassificationTrainer
ImageClassification(
102
ImageClassificationTrainer
.Options options) =>
106
/// Create <see cref="Microsoft.ML.Vision.
ImageClassificationTrainer
"/>, which trains a Deep Neural Network(DNN) to classify images.
121
public static
ImageClassificationTrainer
ImageClassification(