Model

공개 수업 모델

TFLite 모델 및 TFLite 인터프리터를 위한 래퍼 클래스입니다.

참고: Model 한 번에 하나의 TFLite 모델만 보유할 수 있으며 이를 실행하기 위해 항상 TFLite 인터프리터 인스턴스를 보유합니다.

중첩 클래스

수업	모델.빌더		이 클래스는 더 이상 사용되지 않습니다. Model.createModel(Context, String, Options) 사용하십시오.
열거형	모델.장치		분류 실행에 사용되는 런타임 장치 유형입니다.
수업	모델.옵션		모델 실행을 위한 옵션입니다.

공개 방법

무효의	닫다 ()
정적 모델	createModel (컨텍스트 컨텍스트, 문자열 modelPath, Model.Options 옵션) 자산에서 모델을 로드하고 주어진 옵션으로 TFLite 인터프리터를 초기화합니다.
정적 모델	createModel (컨텍스트 컨텍스트, 문자열 modelPath) 자산에서 모델을 로드하고 TFLite 인터프리터를 초기화합니다.
MappedByteBuffer	getData () 메모리 매핑된 모델 데이터를 반환합니다.
텐서	getInputTensor (int inputIndex) 제공된 입력 인덱스와 연결된 Tensor를 가져옵니다.
텐서	getOutputTensor (int 출력 인덱스) 제공된 출력 인덱스와 연결된 Tensor를 가져옵니다.
정수[]	getOutputTensorShape (int 출력 인덱스) 출력 형태를 반환합니다.
끈	getPath () Assets에 저장된 모델 파일의 경로를 반환합니다.
무효의	실행 ( Object[] 입력, Map < Integer , Object > 출력) 여러 입력에 대해 모델 추론을 실행하고 여러 출력을 반환합니다.

상속된 메서드

java.lang.Object 클래스에서

부울	같음 ( 개체 arg0)
최종 클래스 <?>	getClass ()
정수	해시 코드 ()
최종 무효	알림 ()
최종 무효	통지모두 ()
끈	toString ()
최종 무효	대기 (long arg0, int arg1)
최종 무효	기다리세요 (긴 arg0)
최종 무효	기다리다 ()

공개 방법

공공 무효 닫기 ()

공개 정적 모델 createModel (컨텍스트 컨텍스트, 문자열 modelPath, Model.Options 옵션)

자산에서 모델을 로드하고 주어진 옵션으로 TFLite 인터프리터를 초기화합니다.

매개변수

문맥	앱 컨텍스트.
모델경로	모델 파일의 경로입니다.
옵션	모델 실행을 위한 옵션입니다.

던지기

IO예외	모델 파일을 열 때 예외가 발생하는 경우.

또한보십시오

• for details.

공개 정적 모델 createModel (컨텍스트 컨텍스트, 문자열 modelPath)

자산에서 모델을 로드하고 TFLite 인터프리터를 초기화합니다.

기본 옵션은 다음과 같습니다: (1) CPU 장치; (2) 스레드 1개.

매개변수

문맥	앱 컨텍스트.
모델경로	모델 파일의 경로입니다.

던지기

IO예외	모델 파일을 열 때 예외가 발생하는 경우.

공개 MappedByteBuffer getData ()

메모리 매핑된 모델 데이터를 반환합니다.

공개 텐서 getInputTensor (int inputIndex)

제공된 입력 인덱스와 연결된 Tensor를 가져옵니다.

매개변수

입력 인덱스

던지기

IllegalStateException	통역사가 닫혀 있는 경우.

공개 텐서 getOutputTensor (int 출력 인덱스)

제공된 출력 인덱스와 연결된 Tensor를 가져옵니다.

매개변수

출력 인덱스

던지기

IllegalStateException	통역사가 닫혀 있는 경우.

공개 int[] getOutputTensorShape (int 출력 인덱스)

출력 형태를 반환합니다. 그래프가 생성될 때만 출력 형태가 결정되는 경우 유용합니다.

매개변수

출력 인덱스

던지기

IllegalStateException	통역사가 닫혀 있는 경우.

공개 문자열 getPath ()

Assets에 저장된 모델 파일의 경로를 반환합니다.

public void run ( Object[] 입력, Map < Integer , Object > 출력)

여러 입력에 대해 모델 추론을 실행하고 여러 출력을 반환합니다.

매개변수

입력	입력 데이터의 배열. 입력은 모델의 입력과 동일한 순서로 되어 있어야 합니다. 각 입력은 배열 또는 다차원 배열이거나 int, float, long 및 byte를 포함한 기본 유형의 ByteBuffer 일 수 있습니다. ByteBuffer 큰 입력 데이터를 전달하는 데 선호되는 방법인 반면, 문자열 유형은 (다차원) 배열 입력 경로를 사용해야 합니다. ByteBuffer 사용하면 모델 추론이 완료될 때까지 해당 콘텐츠가 변경되지 않은 상태로 유지되어야 합니다.
출력	출력 인덱스를 출력 데이터의 다차원 배열 또는 int, float, long 및 byte를 포함한 기본 유형의 ByteBuffer 에 매핑하는 맵입니다. 사용할 출력에 대한 항목만 유지하면 됩니다.