ML.NET 圖像分類遷移學習可以從保存的模型 .zip 文件或檢查點恢復嗎？ 如何？

Question

我正在使用 NuGet 包 Microsoft.ML (1.4.0) 和 SciSharp.TensorFlow.Redist (1.15.0)

初始訓練和保存模型工作正常

var options = new Microsoft.ML.Vision.ImageClassificationTrainer.Options()
{
    FeatureColumnName = "Image",
    LabelColumnName = "LabelAsKey",
    Arch = Microsoft.ML.Vision.ImageClassificationTrainer.Architecture.InceptionV3,
    Epoch = 50,
    BatchSize = 10,
    LearningRate = 0.01f,
    MetricsCallback = (metrics) => Console.WriteLine(metrics),
    ValidationSet = testDataView
};

var pipeline = mlContext.MulticlassClassification.Trainers.ImageClassification(options)
        .Append(mlContext.Transforms.Conversion.MapKeyToValue(
            outputColumnName: "PredictedLabel", 
            inputColumnName: "PredictedLabel"));


ITransformer trainedModel = pipeline.Fit(trainDataView);

mlContext.Model.Save(trainedModel, trainDataView.Schema, "C:/Temp/model.zip")

並使用模型進行預測工作正常

var loadModel = mlContext.Model.Load("C/Temp/Model.zip", out var modelInputSchema)
var PredictionEngine = mlContext.Model.CreatePredictionEngine<InMemoryImageData, ImagePrediction>(loadModel);

prediction = PredictionEngine.Predict(image);

但是，如果我想繼續用更多圖像訓練 model.zip 文件而不從頭開始重新訓練基線模型，我該怎么做？

有沒有辦法可以再次調用.Fit()使用加載的模型？

Answer 1

ML.NET 不支持恢復對圖像分類訓練器的訓練。 目前，唯一支持它的培訓師如下： https : //docs.microsoft.com/en-us/dotnet/machine-learning/how-to-guides/retrain-model-ml-net

Answer 2

“ML.NET 不支持對圖像分類訓練器進行恢復訓練。目前，唯一支持它的訓練器如下： https : //docs.microsoft.com/en-us/dotnet/machine-learning/how-to -guides/retrain-model-ml-net”

它不支持 LbfgsMaximumEntropyMulticlassTrainer 算法。