ResNet50Extension 类

定义

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

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

方法

ResNet50(DnnImageModelSelector, IHostEnvironment, String, String)

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

ResNet50(DnnImageModelSelector, IHostEnvironment, String, String, String)

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

适用于