Aracılığıyla paylaş


ResNet18Extension Sınıf

Tanım

Bu, önceden eğitilmiş bir ResNet18 modeli kullanmak için ile birlikte DnnImageFeaturizerEstimator kullanılacak bir uzantı yöntemidir. Bu uzantıyı içeren NuGet'in ikili model dosyasını da içermesi garanti edilir.

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

Yöntemler

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

ResNet işlem hattı için gerekli olan iki modele (önişlem ve ana model) sahip bir tahmin zinciri döndürür. Ayrıca rastgele giriş ve çıkış sütun adlarını kullanabilmek için gereken ColumnsCopyingTransforms öğesini yeniden adlandırmayı da içerir. Bu, her iki modelin de bu yöntemi içeren dosyayla aynı konumda olduğunu varsayar ve NuGet aracılığıyla kullanılırsa bunlar olacaktır. Modeli bir NuGet'ten içeri aktarıyorsanız ResNet18'i kullanmanın varsayılan yolu bu olmalıdır.

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

Bu, özel bir model konumunun belirtilmesine olanak tanır. Bu, özel bir modelin belirtilmesi veya modelin ana uygulamadan farklı bir klasöre ayrı olarak yerleştirilmesi veya gönderilmesi isteniyorsa yararlıdır. OnnxTransformer'ın çalışması için ONNX modellerinin tek başına bir dizinde olması gerektiğinden, bu yöntemin her seferinde bu dizini el ile yapmak zorunda kalmak için geçirilen dizine resNet18Onnx/ResNetPrepOnnx alt dizinini eklediğini unutmayın.

Şunlara uygulanır