ResNet18Extension Clase

Definición

Espacio de nombres:

Microsoft.ML

Ensamblado:

Microsoft.ML.DnnImageFeaturizer.ResNet18.dll

Paquete:

Microsoft.ML.DnnImageFeaturizer.ResNet18 v0.21.1

Se trata de un método de extensión que se va a usar con DnnImageFeaturizerEstimator el fin de usar un modelo ResNet18 previamente entrenado. Se garantiza que el paquete NuGet que contiene esta extensión también incluye el archivo de modelo binario.

public static class ResNet18Extension

type ResNet18Extension = class

Public Module ResNet18Extension

Herencia

Object

ResNet18Extension

Métodos

ResNet18(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 ResNet. 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 ResNet18 si importa el modelo desde un NuGet.

ResNet18(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 ONNX deben estar en un directorio por sí mismos para que onnxTransformer funcione, este método anexa un subdirectorio ResNet18Onnx/ResNetPrepOnnx al directorio pasado para evitar tener que realizar ese directorio manualmente cada vez.