AlexNetExtension 类

定义

这是与 DnnImageFeaturizerEstimator 一起使用的扩展方法,以便使用预定型的 AlexNet 模型。 包含此扩展的 NuGet 还保证包含二进制模型文件。

public static class AlexNetExtension
type AlexNetExtension = class
Public Module AlexNetExtension
继承
AlexNetExtension

方法

AlexNet(DnnImageModelSelector, IHostEnvironment, String, String)

返回一个估算器链,其中两个相应的模型 (预处理一个模型,一个是亚历克斯Net 管道所需的主模型) 。 还包括能够使用任意输入和输出列名所需的重命名 ColumnsCopyingTransforms。 这假设这两个模型都与包含此方法的文件位于同一位置,如果通过 NuGet 使用,则使用此方法。 如果从 NuGet 导入模型,则这是使用 AlexNet 的默认方法。

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

这允许指定自定义模型位置。 这是一个有用的自定义模型是指定的,或者,如果需要将模型单独放置或寄送到与主应用程序不同的文件夹中。 请注意,由于 Onnx 模型必须全部位于目录中才能使 OnnxTransformer 正常工作,因此此方法将 AlexNetOnnx/AlexNetPrepOnnx 子目录追加到传入的目录中,以防止每次都手动创建该目录。

适用于