public class
Model
TFLite 모델 및 TFLite 인터프리터의 래퍼 클래스입니다.
참고: Model
는 한 번에 1개의 TFLite 모델만 보유할 수 있으며 모델을 실행할 수 있는 TFLite 인터프리터 인스턴스를 항상 보유합니다.
중첩된 클래스
클래스 | Model.Builder |
이 클래스는 지원 중단되었습니다. Model.createModel(Context, String, Options) 를 사용하세요.
|
|
enum | Model.Device | 분류를 실행하는 데 사용되는 런타임 기기 유형입니다. | |
클래스 | Model.Options | 모델 실행을 위한 옵션입니다. |
공개 메서드
void |
close()
|
정적 모델 |
createModel(컨텍스트 컨텍스트, String modelPath, Model.Options 옵션)
애셋에서 모델을 로드하고 지정된 옵션으로 TFLite 인터프리터를 초기화합니다.
|
정적 모델 | |
MappedByteBuffer |
getData()
메모리 매핑 모델 데이터를 반환합니다.
|
Tensor |
getInputTensor(int inputIndex)
제공된 입력 색인과 연결된 텐서를 가져옵니다.
|
Tensor |
getOutputTensor(int outputIndex)
제공된 출력 색인과 연결된 텐서를 가져옵니다.
|
int[] |
getOutputTensorShape(int outputIndex)
출력 셰이프를 반환합니다.
|
String |
getPath()
애셋에 저장된 모델 파일의 경로를 반환합니다.
|
void |
상속된 메서드
공개 메서드
public void close ()
public static 모델 createModel (Context context, String modelPath, Model.Options options)
애셋에서 모델을 로드하고 지정된 옵션으로 TFLite 인터프리터를 초기화합니다.
매개변수
context | 앱 컨텍스트 |
---|---|
modelPath | 모델 파일의 경로입니다. |
옵션 | 모델 실행을 위한 옵션입니다. |
생성 값
IOException | 모델 파일을 열 때 예외가 발생하는 경우 |
---|
더보기
public static 모델 createModel (Context context, String modelPath)
애셋에서 모델을 로드하고 TFLite 인터프리터를 초기화합니다.
기본 옵션은 (1) CPU 기기, (2) 스레드 1개입니다.
매개변수
context | 앱 컨텍스트 |
---|---|
modelPath | 모델 파일의 경로입니다. |
생성 값
IOException | 모델 파일을 열 때 예외가 발생하는 경우 |
---|
public Tensor getInputTensor (int inputIndex)
public Tensor getOutputTensor (int outputIndex)
public int[] getOutputTensorShape (int outputIndex)
출력 셰이프를 반환합니다. 그래프를 생성할 때만 출력 형태가 결정되는 경우에 유용합니다.
매개변수
outputIndex |
---|
생성 값
IllegalStateException | 닫을 수 있습니다 |
---|
public void run (Object[] 입력, 매핑<정수, 객체> 출력)
여러 입력에 대해 모델 추론을 실행하고 여러 출력을 반환합니다.
매개변수
입력 | 입력 데이터의 배열입니다. 입력의 순서는 모델 입력과 동일해야 합니다. 각 입력은 배열 또는 다차원 배열이거나 기본형(int, float, long, byte)의 ByteBuffer 일 수 있습니다. ByteBuffer 는 대규모 입력 데이터를 전달하는 데 선호되는 방법이고, 문자열 유형에는 다차원 배열 입력 경로를 사용해야 합니다. ByteBuffer 가 사용되는 경우 모델 추론이 완료될 때까지 콘텐츠가 변경되지 않아야 합니다. |
---|---|
출력 | 출력 색인을 다차원 배열로 매핑하거나 int, float, long, byte를 비롯한 원시 유형의 ByteBuffer 에 매핑합니다. 항목을 유지하기만 하면 출력을 사용할 수 있습니다.
|