ResNet18Extension Classe

Definição

Namespace:

Microsoft.ML

Assembly:

Microsoft.ML.DnnImageFeaturizer.ResNet18.dll

Pacote:

Microsoft.ML.DnnImageFeaturizer.ResNet18 v0.21.1

Esse é um método de extensão a ser usado com o DnnImageFeaturizerEstimator modelo ResNet18 pré-treinado. O NuGet que contém essa extensão também garante a inclusão do arquivo de modelo binário.

public static class ResNet18Extension

type ResNet18Extension = class

Public Module ResNet18Extension

Herança

Object

ResNet18Extension

Métodos

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

Retorna uma cadeia de estimativas com os dois modelos correspondentes (um pré-processamento e um principal) necessários para o pipeline resNet. Também inclui a renomeação columnsCopyingTransforms necessária para poder usar nomes arbitrários de coluna de entrada e saída. Isso pressupõe que ambos os modelos estejam no mesmo local que o arquivo que contém esse método, que eles serão se usados por meio do NuGet. Essa deve ser a maneira padrão de usar ResNet18 se importar o modelo de um NuGet.

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

Isso permite que um local de modelo personalizado seja especificado. Isso é útil é que um modelo personalizado é especificado ou se o modelo deseja ser colocado ou enviado separadamente em uma pasta diferente do aplicativo principal. Observe que, como os modelos ONNX devem estar em um diretório sozinhos para que o OnnxTransformer funcione, esse método acrescenta um subdiretório ResNet18Onnx/ResNetPrepOnnx ao diretório passado para evitar a necessidade de fazer esse diretório manualmente a cada vez.