AlexNetExtension Classe

Definição

Namespace:

Microsoft.ML

Assembly:

Microsoft.ML.DnnImageFeaturizer.AlexNet.dll

Pacote:

Microsoft.ML.DnnImageFeaturizer.AlexNet v0.21.1

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

public static class AlexNetExtension

type AlexNetExtension = class

Public Module AlexNetExtension

Herança

Object

AlexNetExtension

Métodos

AlexNet(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 AlexNet. 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 AlexNet se importar o modelo de um NuGet.

AlexNet(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 AlexNetOnnx/AlexNetPrepOnnx ao diretório passado para evitar a necessidade de fazer esse diretório manualmente a cada vez.