Condividi tramite


AlexNetExtension Classe

Definizione

Metodo di estensione da usare con DnnImageFeaturizerEstimator per poter usare un modello AlexNet con pre-training. Il pacchetto NuGet contenente questa estensione include anche il file di modello binario.

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

Metodi

AlexNet(DnnImageModelSelector, IHostEnvironment, String, String)

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

AlexNet(DnnImageModelSelector, IHostEnvironment, String, String, String)

Ciò consente di specificare una posizione del modello personalizzata. Questo è utile è un modello personalizzato specificato oppure se il modello deve essere inserito o inviato separatamente in una cartella diversa dall'applicazione principale. Si noti che poiché i modelli Onnx devono trovarsi in una directory da soli per il funzionamento di OnnxTransformer, questo metodo aggiunge una sottodirectory AlexNetOnnx/AlexNetPrepOnnx alla directory passata per evitare di dover eseguire manualmente tale directory ogni volta.

Si applica a