공개 수업 모델
TFLite 모델 및 TFLite 인터프리터를 위한 래퍼 클래스입니다.
참고: Model
한 번에 하나의 TFLite 모델만 보유할 수 있으며 이를 실행하기 위해 항상 TFLite 인터프리터 인스턴스를 보유합니다.
중첩 클래스
수업 | 모델.빌더 | 이 클래스는 더 이상 사용되지 않습니다. Model.createModel(Context, String, Options) 사용하십시오. | |
열거형 | 모델.장치 | 분류 실행에 사용되는 런타임 장치 유형입니다. | |
수업 | 모델.옵션 | 모델 실행을 위한 옵션입니다. |
공개 방법
무효의 | 닫다 () |
정적 모델 | createModel (컨텍스트 컨텍스트, 문자열 modelPath, Model.Options 옵션) 자산에서 모델을 로드하고 주어진 옵션으로 TFLite 인터프리터를 초기화합니다. |
정적 모델 | |
MappedByteBuffer | getData () 메모리 매핑된 모델 데이터를 반환합니다. |
텐서 | getInputTensor (int inputIndex) 제공된 입력 인덱스와 연결된 Tensor를 가져옵니다. |
텐서 | getOutputTensor (int 출력 인덱스) 제공된 출력 인덱스와 연결된 Tensor를 가져옵니다. |
정수[] | getOutputTensorShape (int 출력 인덱스) 출력 형태를 반환합니다. |
끈 | getPath () Assets에 저장된 모델 파일의 경로를 반환합니다. |
무효의 |
상속된 메서드
공개 방법
공공 무효 닫기 ()
공개 정적 모델 createModel (컨텍스트 컨텍스트, 문자열 modelPath, Model.Options 옵션)
자산에서 모델을 로드하고 주어진 옵션으로 TFLite 인터프리터를 초기화합니다.
매개변수
문맥 | 앱 컨텍스트. |
---|---|
모델경로 | 모델 파일의 경로입니다. |
옵션 | 모델 실행을 위한 옵션입니다. |
던지기
IO예외 | 모델 파일을 열 때 예외가 발생하는 경우. |
---|
또한보십시오
공개 정적 모델 createModel (컨텍스트 컨텍스트, 문자열 modelPath)
자산에서 모델을 로드하고 TFLite 인터프리터를 초기화합니다.
기본 옵션은 다음과 같습니다: (1) CPU 장치; (2) 스레드 1개.
매개변수
문맥 | 앱 컨텍스트. |
---|---|
모델경로 | 모델 파일의 경로입니다. |
던지기
IO예외 | 모델 파일을 열 때 예외가 발생하는 경우. |
---|
공개 텐서 getInputTensor (int inputIndex)
공개 텐서 getOutputTensor (int 출력 인덱스)
공개 int[] getOutputTensorShape (int 출력 인덱스)
출력 형태를 반환합니다. 그래프가 생성될 때만 출력 형태가 결정되는 경우 유용합니다.
매개변수
출력 인덱스 |
---|
던지기
IllegalStateException | 통역사가 닫혀 있는 경우. |
---|
public void run ( Object[] 입력, Map < Integer , Object > 출력)
여러 입력에 대해 모델 추론을 실행하고 여러 출력을 반환합니다.
매개변수
입력 | 입력 데이터의 배열. 입력은 모델의 입력과 동일한 순서로 되어 있어야 합니다. 각 입력은 배열 또는 다차원 배열이거나 int, float, long 및 byte를 포함한 기본 유형의 ByteBuffer 일 수 있습니다. ByteBuffer 큰 입력 데이터를 전달하는 데 선호되는 방법인 반면, 문자열 유형은 (다차원) 배열 입력 경로를 사용해야 합니다. ByteBuffer 사용하면 모델 추론이 완료될 때까지 해당 콘텐츠가 변경되지 않은 상태로 유지되어야 합니다. |
---|---|
출력 | 출력 인덱스를 출력 데이터의 다차원 배열 또는 int, float, long 및 byte를 포함한 기본 유형의 ByteBuffer 에 매핑하는 맵입니다. 사용할 출력에 대한 항목만 유지하면 됩니다. |