lớp công khai LabelUtil
Tiện ích hoạt động nhãn.
Phương pháp công khai
Danh sách tĩnh < Chuỗi > | mapValueToLabels ( TensorBuffer tensorBuffer, List < String > labels, int offset) Ánh xạ một tenxơ giá trị int vào danh sách các nhãn chuỗi. |
Phương pháp kế thừa
Phương pháp công khai
Danh sách tĩnh công khai < Chuỗi > mapValueToLabels ( TensorBuffer tensorBuffer, Danh sách < Chuỗi > nhãn, int offset)
Ánh xạ một tenxơ giá trị int vào danh sách các nhãn chuỗi. Nó lấy một chuỗi các chuỗi làm từ điển. Ví dụ: nếu tensor đã cho là [3, 1, 0] và các nhãn đã cho là ["background", "apple", "banana", "cherry", "date"] thì kết quả sẽ là ["date", "chuối", "táo"].
Thông số
bộ đệm tensor | Một tensor với các giá trị chỉ số. Các giá trị phải là số nguyên không âm và mỗi giá trị x sẽ được chuyển đổi thành labels[x + offset] . Nếu tenxơ được đưa ra dưới dạng float TensorBuffer , các giá trị sẽ được chuyển thành số nguyên. Tất cả các giá trị nằm ngoài giới hạn sẽ ánh xạ tới chuỗi trống. |
---|---|
nhãn | Danh sách các chuỗi, dùng làm từ điển để tra cứu. Chỉ mục của phần tử mảng sẽ được sử dụng làm khóa. Để có hiệu suất tốt hơn, hãy sử dụng đối tượng triển khai RandomAccess, chẳng hạn như ArrayList . |
bù lại | Giá trị offset khi tra cứu giá trị int trong labels . |
Trả lại
- các chuỗi được ánh xạ. Độ dài của danh sách là
TensorBuffer.getFlatSize()
.
Ném
Ngoại lệ Đối số bất hợp pháp | nếu tensorBuffer hoặc labels là null. |
---|