다음을 통해 공유


ResNet18Extension 클래스

정의

미리 학습된 ResNet18 모델을 사용하기 위해 사용할 DnnImageFeaturizerEstimator 확장 메서드입니다. 이 확장을 포함하는 NuGet도 이진 모델 파일을 포함하도록 보장됩니다.

public static class ResNet18Extension
type ResNet18Extension = class
Public Module ResNet18Extension
상속
ResNet18Extension

메서드

ResNet18(DnnImageModelSelector, IHostEnvironment, String, String)

ResNet 파이프라인에 필요한 두 개의 해당 모델(전처리 모델과 기본 모델)이 있는 추정기 체인을 반환합니다. 또한 임의의 입력 및 출력 열 이름을 사용할 수 있도록 하는 데 필요한 ColumnsCopyingTransform 이름 바꾸기도 포함됩니다. 두 모델이 모두 이 메서드를 포함하는 파일과 동일한 위치에 있다고 가정합니다. 이 위치는 NuGet을 통해 사용되는 경우입니다. NuGet에서 모델을 가져오는 경우 ResNet18을 사용하는 기본 방법이어야 합니다.

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

이렇게 하면 사용자 지정 모델 위치를 지정할 수 있습니다. 이는 사용자 지정 모델을 지정하거나 모델을 주 애플리케이션과 다른 폴더에 별도로 배치하거나 배송하려는 경우에 유용합니다. OnnxTransformer가 작동하려면 ONNX 모델이 모두 디렉터리에 있어야 하므로 이 메서드는 매번 해당 디렉터리를 수동으로 만들 필요가 없도록 전달된 디렉터리에 ResNet18Onnx/ResNetPrepOnnx 하위 디렉터리를 추가합니다.

적용 대상