Freigeben über


ResNet18Extension Klasse

Definition

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
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.

Gilt für: