ResNet50Extension Classe

Definição

Esse é um método de extensão a ser usado para DnnImageFeaturizerEstimator usar um modelo ResNet50 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 ResNet50Extension
Herança
ResNet50Extension

Métodos

ResNet50(DnnImageModelSelector, IHostEnvironment, String, String)

Retorna uma cadeia de avaliadores com os dois modelos correspondentes (um pré-processamento e um principal) necessários para o pipeline resNet. Também inclui a renomeação de ColumnsCopyingTransforms necessárias 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 serão usados por meio do NuGet. Essa deve ser a maneira padrão de usar ResNet50 se importar o modelo de um NuGet.

ResNet50(DnnImageModelSelector, IHostEnvironment, String, String, String)

Isso permite que um local de modelo personalizado seja especificado. Isso é útil se um modelo personalizado for especificado ou se o modelo desejar 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 ResNet50Onnx/ResNetPrepOnnx ao diretório passado para evitar a necessidade de fazer esse diretório manualmente a cada vez.

Aplica-se a