ResNet18Extension Classe

Définition

Espace de noms:

Microsoft.ML

Assembly:

Microsoft.ML.DnnImageFeaturizer.ResNet18.dll

Paquet:

Microsoft.ML.DnnImageFeaturizer.ResNet18 v0.21.1

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

Object

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.