Bagikan melalui


ResNet18Extension Kelas

Definisi

Ini adalah metode ekstensi yang akan digunakan dengan DnnImageFeaturizerEstimator untuk menggunakan model ResNet18 yang telah dilatih sebelumnya. NuGet yang berisi ekstensi ini juga dijamin untuk menyertakan file model biner.

public static class ResNet18Extension
type ResNet18Extension = class
Public Module ResNet18Extension
Warisan
ResNet18Extension

Metode

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

Mengembalikan rantai estimator dengan dua model yang sesuai (praproses dan yang utama) yang diperlukan untuk alur ResNet. Juga termasuk penggantian nama ColumnsCopyingTransforms yang diperlukan untuk dapat menggunakan nama kolom input dan output arbitrer. Ini mengasumsikan kedua model berada di lokasi yang sama dengan file yang berisi metode ini, yang akan jika digunakan melalui NuGet. Ini harus menjadi cara default untuk menggunakan ResNet18 jika mengimpor model dari NuGet.

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

Ini memungkinkan lokasi model kustom ditentukan. Ini berguna adalah model kustom ditentukan, atau jika model diinginkan untuk ditempatkan atau dikirim secara terpisah di folder yang berbeda dari aplikasi utama. Perhatikan bahwa karena model ONNX harus berada dalam direktori sendiri agar OnnxTransformer berfungsi, metode ini menambahkan subdirektori ResNet18Onnx/ResNetPrepOnnx ke direktori yang diteruskan untuk mencegah keharusan membuat direktori tersebut secara manual setiap kali.

Berlaku untuk