Condividi tramite


ResNet18Extension Classe

Definizione

Si tratta di un metodo di estensione da usare con DnnImageFeaturizerEstimator per usare un modello ResNet18 con training preliminare. Il pacchetto NuGet contenente questa estensione include anche il file di modello binario.

public static class ResNet18Extension
type ResNet18Extension = class
Public Module ResNet18Extension
Ereditarietà
ResNet18Extension

Metodi

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

Restituisce una catena di stima con i due modelli corrispondenti (uno di pre-elaborazione e uno principale) necessari per la pipeline ResNet. Include anche la ridenominazione di ColumnsCopyingTransforms necessaria per poter usare nomi arbitrari di input e di colonna di output. Si presuppone che entrambi i modelli si trovino nella stessa posizione del file contenente questo metodo, che verranno usati tramite NuGet. Questo dovrebbe essere il modo predefinito per usare ResNet18 se si importa il modello da un NuGet.

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

In questo modo è possibile specificare una posizione del modello personalizzata. Ciò è utile è un modello personalizzato specificato oppure se si desidera inserire o spedire separatamente il modello in una cartella diversa dall'applicazione principale. Si noti che poiché i modelli ONNX devono trovarsi in una directory per il corretto funzionamento di OnnxTransformer, questo metodo aggiunge una sottodirectory ResNet18Onnx/ResNetPrepOnnx alla directory passata per evitare di dover eseguire manualmente tale directory ogni volta.

Si applica a