ResNet18Extension 类

定义

命名空间:

Microsoft.ML

程序集:

Microsoft.ML.DnnImageFeaturizer.ResNet18.dll

包:

Microsoft.ML.DnnImageFeaturizer.ResNet18 v0.21.1

这是一种扩展方法 DnnImageFeaturizerEstimator ，用于使用预先训练的 ResNet18 模型。 包含此扩展的 NuGet 还保证包含二进制模型文件。

public static class ResNet18Extension

type ResNet18Extension = class

Public Module ResNet18Extension

继承

Object

ResNet18Extension

方法

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)	返回一个估算器链，其中两个相应的模型 (预处理一个模型，一个是 ResNet 管道所需的主模型) 。 还包括能够使用任意输入和输出列名所需的重命名 ColumnsCopyingTransforms。 这假设这两个模型都与包含此方法的文件位于同一位置，如果通过 NuGet 使用，则使用此方法。 这是从 NuGet 导入模型时使用 ResNet18 的默认方法。
ResNet18(DnnImageModelSelector, IHostEnvironment, String, String, String)	这允许指定自定义模型位置。 这是一个有用的自定义模型是指定的，或者，如果需要将模型单独放置或寄送到与主应用程序不同的文件夹中。 请注意，由于 ONNX 模型必须全部位于目录中才能使 OnnxTransformer 正常工作，因此此方法将 ResNet18Onnx/ResNetPrepOnnx 子目录追加到传入的目录中，以防止每次都必须手动创建该目录。