이미지에 대한 분류를 수행합니다.
API는 선택 사항이지만 강력히 권장되는 TFLite 모델 메타데이터가 있는 TFLite 모델을 기대합니다. .
API는 하나의 이미지 입력 텐서와 하나의 분류 출력 텐서가 있는 모델을 지원합니다. 보다 구체적으로 요구 사항은 다음과 같습니다.
- 입력 이미지 텐서(
kTfLiteUInt8
/kTfLiteFloat32
)-
[batch x height x width x channels]
크기의 이미지 입력. - 일괄 추론은 지원되지 않습니다(
batch
는 1이어야 함). - RGB 입력만 지원됩니다(
channels
3개여야 함). - 유형이
kTfLiteFloat32
이면 입력 정규화를 위해 메타데이터에 NormalizationOptions를 연결해야 합니다.
-
kTfLiteUInt8
/ kTfLiteFloat32
)-
[1 x N]
또는[1 x 1 x 1 x N]
과 같이 2차원 또는 4차원의N
클래스 포함 - 레이블 파일은 메타데이터에 압축되어야 합니다. 이미지 분류자에 대한 메타데이터 생성 예를 참조하세요. 레이블 파일이 압축되지 않은 경우 결과에서 인덱스를 레이블로 사용합니다.
이러한 모델의 예는 TensorFlow Hub에서 찾을 수 있습니다. .
중첩 클래스
수업 | ImageClassifier.ImageClassifier옵션 | ImageClassifier 설정 옵션. |
공개 방법
상속된 메서드
공개 방법
공개 목록 < 분류 > 분류 ( TensorImage 이미지)
제공된 TensorImage
에 대한 실제 분류를 수행합니다.
ImageClassifier
다음 TensorImage
색상 공간 유형을 지원합니다.
매개변수
영상 | RGB 또는 YUV 이미지를 나타내는 UINT8 TensorImage 객체 |
---|
던지기
IllegalArgumentException | 이미지의 색상 공간 유형이 지원되지 않는 경우 |
---|
공개 목록 < 분류 > 분류 ( TensorImage 이미지, ImageProcessingOptions 옵션)
ImageProcessingOptions
사용하여 제공된 TensorImage
에 대한 실제 분류를 수행합니다.
ImageClassifier
다음 옵션을 지원합니다.
- 관심 영역(ROI)(
ImageProcessingOptions.Builder.setRoi(Rect)
통해) 기본값은 전체 이미지입니다. - 이미지 회전(
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
을 통해). 기본값은ImageProcessingOptions.Orientation.TOP_LEFT
입니다.
ImageClassifier
다음 TensorImage
색상 공간 유형을 지원합니다.
매개변수
영상 | RGB 또는 YUV 이미지를 나타내는 UINT8 TensorImage 객체 |
---|---|
옵션 |
던지기
IllegalArgumentException | 이미지의 색상 공간 유형이 지원되지 않는 경우 |
---|
공개 목록 < 분류 > 분류 ( MlImage 이미지)
제공된 MlImage
에 대한 실제 분류를 수행합니다.
매개변수
영상 | 이미지를 나타내는 MlImage 객체 |
---|
던지기
IllegalArgumentException | 이미지의 저장 유형이나 형식이 지원되지 않는 경우 |
---|
공개 목록 < 분류 > 분류 ( MlImage 이미지, ImageProcessingOptions 옵션)
ImageProcessingOptions
사용하여 제공된 MlImage
에 대한 실제 분류를 수행합니다.
ImageClassifier
다음 옵션을 지원합니다.
- 관심 영역(ROI)(
ImageProcessingOptions.Builder.setRoi(Rect)
통해) 기본값은 전체 이미지입니다. - 이미지 회전(
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
을 통해). 기본값은ImageProcessingOptions.Orientation.TOP_LEFT
입니다.MlImage.getRotation()
이 효과적이지 않습니다.
매개변수
영상 | 이미지를 나타내는 MlImage 객체 |
---|---|
옵션 | ROI 및 회전을 포함한 옵션 구성 |
던지기
IllegalArgumentException | 이미지의 저장 유형이나 형식이 지원되지 않는 경우 |
---|
공개 정적 ImageClassifier createFromBuffer ( ByteBuffer modelBuffer)
모델 버퍼와 기본 ImageClassifier.ImageClassifierOptions
사용하여 ImageClassifier
인스턴스를 생성합니다.
매개변수
모델버퍼 | 직접 ByteBuffer 또는 분류 모델의 MappedByteBuffer |
---|
던지기
IllegalArgumentException | 모델 버퍼가 직접 ByteBuffer 또는 MappedByteBuffer 가 아닌 경우 |
---|---|
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
공개 정적 ImageClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions 옵션)
모델 버퍼와 ImageClassifier.ImageClassifierOptions
사용하여 ImageClassifier
인스턴스를 생성합니다.
매개변수
모델버퍼 | 직접 ByteBuffer 또는 분류 모델의 MappedByteBuffer |
---|---|
옵션 |
던지기
IllegalArgumentException | 모델 버퍼가 직접 ByteBuffer 또는 MappedByteBuffer 가 아닌 경우 |
---|---|
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
공개 정적 ImageClassifier createFromFile (컨텍스트 컨텍스트, 문자열 modelPath)
기본 ImageClassifier.ImageClassifierOptions
에서 ImageClassifier
인스턴스를 만듭니다.
매개변수
문맥 | |
---|---|
모델경로 | 자산의 메타데이터가 있는 분류 모델의 경로 |
던지기
IO예외 | tflite 모델을 로드할 때 I/O 오류가 발생하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
공개 정적 ImageClassifier createFromFile ( 파일 모델파일)
기본 ImageClassifier.ImageClassifierOptions
에서 ImageClassifier
인스턴스를 만듭니다.
매개변수
모델파일 | 분류 모델 File 인스턴스 |
---|
던지기
IO예외 | tflite 모델을 로드할 때 I/O 오류가 발생하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
공개 정적 ImageClassifier createFromFileAndOptions (컨텍스트 컨텍스트, 문자열 modelPath, ImageClassifier.ImageClassifierOptions 옵션)
ImageClassifier.ImageClassifierOptions
에서 ImageClassifier
인스턴스를 만듭니다.
매개변수
문맥 | |
---|---|
모델경로 | 자산의 메타데이터가 있는 분류 모델의 경로 |
옵션 |
던지기
IO예외 | tflite 모델을 로드할 때 I/O 오류가 발생하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |
public static ImageClassifier createFromFileAndOptions ( 파일 모델파일, ImageClassifier.ImageClassifierOptions 옵션)
ImageClassifier
인스턴스를 생성합니다.
매개변수
모델파일 | 분류 모델 File 인스턴스 |
---|---|
옵션 |
던지기
IO예외 | tflite 모델을 로드할 때 I/O 오류가 발생하는 경우 |
---|---|
IllegalArgumentException | 인수가 유효하지 않은 경우 |
IllegalStateException | 내부 오류가 있는 경우 |
런타임예외 | 달리 명시되지 않은 오류가 있는 경우 |