ResNet18Extension 類別

定義

這是要搭配 DnnImageFeaturizerEstimator 使用的擴充方法,以便使用預先定型的 ResNet18 模型。 包含此延伸模組的 NuGet 也一定會包含二進位模型檔。

public static class ResNet18Extension
type ResNet18Extension = class
Public Module ResNet18Extension
繼承
ResNet18Extension

方法

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

傳回估算器鏈結,其中兩個對應的模型 (前置處理,以及 ResNet 管線所需的主要模型) 。 此外,還包含重新命名 ColumnsCopyingTransforms,才能使用任意輸入和輸出資料行名稱。 這假設這兩個模型都位於與包含此方法之檔案相同的位置,如果透過 NuGet 使用,則會使用它們。 如果從 NuGet 匯入模型,這應該是使用 ResNet18 的預設方式。

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String, String)

這可讓指定自訂模型位置。 這很實用,是指定自訂模型,或者,如果想要將模型放在主應用程式的不同資料夾中或個別出貨,則為 。 請注意,因為 ONNX 模型必須單獨位於目錄中,OnnxTransformer 才能運作,所以此方法會將 ResNet18Onnx/ResNetPrepOnnx 子目錄附加至傳入目錄,以避免每次都必須手動建立該目錄。

適用於