Compartir a través de


AlexNetExtension Clase

Definición

Se trata de un método de extensión que se usará con DnnImageFeaturizerEstimator para poder utilizar un modelo AlexNet previamente entrenado. Se garantiza que el paquete NuGet que contiene esta extensión también incluye el archivo de modelo binario.

public static class AlexNetExtension
type AlexNetExtension = class
Public Module AlexNetExtension
Herencia
AlexNetExtension

Métodos

AlexNet(DnnImageModelSelector, IHostEnvironment, String, String)

Devuelve una cadena de estimador con los dos modelos correspondientes (un preprocesamiento uno y uno principal) necesario para la canalización de AlexNet. También incluye el cambio de nombre ColumnsCopyingTransforms necesario para poder usar nombres arbitrarios de columnas de entrada y salida. Se supone que ambos modelos están en la misma ubicación que el archivo que contiene este método, que se usarán a través de NuGet. Esta debe ser la manera predeterminada de usar AlexNet si importa el modelo desde un NuGet.

AlexNet(DnnImageModelSelector, IHostEnvironment, String, String, String)

Esto permite especificar una ubicación de modelo personalizada. Esto es útil es especificar un modelo personalizado o si se desea colocar o enviar por separado el modelo en una carpeta diferente de la aplicación principal. Tenga en cuenta que, dado que los modelos de Onnx deben estar en un directorio por sí mismos para que OnnxTransformer funcione, este método anexa un subdirectorio AlexNetOnnx/AlexNetPrepOnnx al directorio pasado para evitar tener que hacer ese directorio manualmente cada vez.

Se aplica a