Partager via


ResNet18Extension Classe

Définition

Il s’agit d’une méthode d’extension à utiliser pour DnnImageFeaturizerEstimator utiliser un modèle ResNet18 préentraîné. Le package NuGet contenant cette extension est également garanti pour inclure le fichier modèle binaire.

public static class ResNet18Extension
type ResNet18Extension = class
Public Module ResNet18Extension
Héritage
ResNet18Extension

Méthodes

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

Retourne une chaîne d’estimateur avec les deux modèles correspondants (un prétraitement un et un principal) requis pour le pipeline ResNet. Inclut également le renommage de ColumnsCopyingTransforms requis pour pouvoir utiliser des noms de colonnes d’entrée et de sortie arbitraires. Cela suppose que les deux modèles se trouvent dans le même emplacement que le fichier contenant cette méthode, qu’ils seront utilisés par le biais du NuGet. Il doit s’agir du moyen par défaut d’utiliser ResNet18 si vous importez le modèle à partir d’un NuGet.

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

Cela permet de spécifier un emplacement de modèle personnalisé. Il s’agit d’un modèle personnalisé spécifié, ou si le modèle doit être placé ou expédié séparément dans un dossier différent de l’application principale. Notez que, étant donné que les modèles ONNX doivent se trouver dans un répertoire tout seul pour que l’onnxTransformer fonctionne, cette méthode ajoute un sous-répertoire ResNet18Onnx/ResNetPrepOnnx au répertoire passé pour éviter d’avoir à rendre ce répertoire manuellement chaque fois.

S’applique à