ResNet18Extension Klasse

Definition

Namespace:

Microsoft.ML

Assembly:

Microsoft.ML.DnnImageFeaturizer.ResNet18.dll

Paket:

Microsoft.ML.DnnImageFeaturizer.ResNet18 v0.21.1

Dies ist eine Erweiterungsmethode, die für DnnImageFeaturizerEstimator die Verwendung eines vorab trainierten ResNet18-Modells verwendet werden soll. Das NuGet-Paket, das diese Erweiterung enthält, beinhaltet außerdem garantiert die binäre Modelldatei.

public static class ResNet18Extension

type ResNet18Extension = class

Public Module ResNet18Extension

Vererbung

Object

ResNet18Extension

Methoden

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

Gibt eine Stimatorkette mit den beiden entsprechenden Modellen (einer Vorverarbeitung und einer Hauptkette) zurück, die für die ResNet-Pipeline erforderlich ist. Enthält auch die Umbenennen von ColumnCopyingTransforms, die erforderlich sind, um beliebige Eingabe- und Ausgabespaltennamen zu verwenden. Dies geht davon aus, dass beide Modelle an demselben Speicherort liegen wie die Datei, die diese Methode enthält, die sie durch das NuGet verwenden werden. Dies sollte die standardweise Verwendung von ResNet18 sein, wenn das Modell aus einem NuGet importiert wird.

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String, String)

Dadurch kann ein benutzerdefinierter Modellspeicherort angegeben werden. Dies ist nützlich, wenn ein benutzerdefiniertes Modell angegeben wird, oder wenn das Modell separat in einem anderen Ordner aus der Hauptanwendung platziert oder gesendet werden soll. Beachten Sie, dass ONNX-Modelle sich in einem Verzeichnis befinden müssen, damit onnxTransformer funktioniert, fügt diese Methode jedes Mal ein ResNet18Onnx/ResNetPrepOnnx-Unterverzeichnis an das übergebene Verzeichnis an, um zu verhindern, dass dieses Verzeichnis jedes Mal manuell vorgenommen werden muss.