LabelUtil

공개 클래스 LabelUtil

라벨 작업 유틸리티.

공개 방법

정적 목록 < 문자열 >
mapValueToLabels ( TensorBuffer tensorBuffer, List < String > labels, int offset)
int 값 텐서를 문자열 레이블 목록에 매핑합니다.

상속된 메서드

공개 방법

공개 정적 List < String > mapValueToLabels ( TensorBuffer tensorBuffer, List < String > labels, int offset)

int 값 텐서를 문자열 레이블 목록에 매핑합니다. 문자열 배열을 사전으로 사용합니다. 예: 주어진 텐서가 [3, 1, 0]이고 주어진 레이블이 ["배경", "사과", "바나나", "체리", "날짜"]인 경우 결과는 ["날짜", "바나나", "사과"].

매개변수
텐서버퍼 인덱스 값이 있는 텐서. 값은 음수가 아닌 정수여야 하며 각 값 x labels[x + offset] 으로 변환됩니다. 텐서가 float TensorBuffer 로 제공되면 값은 정수로 변환됩니다. 범위를 벗어난 모든 값은 빈 문자열에 매핑됩니다.
라벨 조회를 위해 사전으로 사용되는 문자열 목록입니다. 배열 요소의 인덱스가 키로 사용됩니다. 더 나은 성능을 얻으려면 ArrayList 와 같이 RandomAccess를 구현하는 객체를 사용하십시오.
오프셋 labels 에서 int 값을 조회할 때의 오프셋 값입니다.
보고
던지기
IllegalArgumentException tensorBuffer 또는 labels 가 null인 경우.