ResNet101Extension 类

定义

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

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

方法

ResNet101(DnnImageModelSelector, IHostEnvironment, String, String)

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

ResNet101(DnnImageModelSelector, IHostEnvironment, String, String, String)

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

适用于