Klasa opakowania dla modelu TFLite i interpretera TFLite.
Uwaga: Model
może przechowywać jednocześnie tylko 1 model TFLite i zawsze przechowuje instancję interpretera TFLite, aby go uruchomić.
Klasy zagnieżdżone
klasa | Model.Konstruktor | Ta klasa jest przestarzała. Proszę użyć Model.createModel(Context, String, Options) . | |
wyliczenie | Model.Urządzenie | Typ urządzenia wykonawczego używanego do wykonywania klasyfikacji. | |
klasa | Model.Opcje | Opcje uruchamiania modelu. |
Metody publiczne
próżnia | zamknąć () |
Model statyczny | createModel (kontekst kontekstowy, string modelPath, opcje Model.Options ) Ładuje model z zasobów i inicjuje interpreter TFLite z podanymi opcjami. |
Model statyczny | createModel (kontekst kontekstowy, String modelPath) Ładuje model z zasobów i inicjuje interpreter TFLite. |
MappedByteBuffer | otrzymać dane () Zwraca dane modelu odwzorowanego w pamięci. |
Napinacz | getInputTensor (int inputIndex) Pobiera Tensor skojarzony z podanym indeksem wejściowym. |
Napinacz | getOutputTensor (int wynikIndex) Pobiera Tensor skojarzony z podanym indeksem wyjściowym. |
int[] | getOutputTensorShape (int wynikIndex) Zwraca kształt wyjściowy. |
Strunowy | getPath () Zwraca ścieżkę pliku modelu przechowywanego w Assets. |
próżnia | run ( obiekt [] wejścia, mapa < liczba całkowita , obiekt > wyjścia) Uruchamia wnioskowanie modelu na wielu danych wejściowych i zwraca wiele wyników. |
Metody dziedziczone
Metody publiczne
publiczna pustka zamknij ()
public static Model createModel (kontekst kontekstowy, String modelPath, opcje Model.Options )
Ładuje model z zasobów i inicjuje interpreter TFLite z podanymi opcjami.
Parametry
kontekst | Kontekst aplikacji. |
---|---|
Ścieżka modelu | Ścieżka pliku modelu. |
opcje | Opcje uruchomienia modelu. |
Rzuca
Wyjątek IO | jeśli wystąpi jakikolwiek wyjątek podczas otwierania pliku modelu. |
---|
Zobacz też
publiczny model statyczny createModel (kontekst kontekstowy, ciąg modelPath)
Ładuje model z zasobów i inicjuje interpreter TFLite.
Domyślne opcje to: (1) urządzenie CPU; (2) jeden wątek.
Parametry
kontekst | Kontekst aplikacji. |
---|---|
Ścieżka modelu | Ścieżka pliku modelu. |
Rzuca
Wyjątek IO | jeśli wystąpi jakikolwiek wyjątek podczas otwierania pliku modelu. |
---|
publiczny Tensor getInputTensor (int inputIndex)
Pobiera Tensor skojarzony z podanym indeksem wejściowym.
Parametry
Indeks wejściowy |
---|
Rzuca
Wyjątek IllegalStateException | jeśli tłumacz jest zamknięty. |
---|
publiczny Tensor getOutputTensor (int OutputIndex)
Pobiera Tensor skojarzony z podanym indeksem wyjściowym.
Parametry
indeks wyjściowy |
---|
Rzuca
Wyjątek IllegalStateException | jeśli tłumacz jest zamknięty. |
---|
public int[] getOutputTensorShape (int wynikIndex)
Zwraca kształt wyjściowy. Przydatne, jeśli kształt wyjściowy jest określany dopiero podczas tworzenia wykresu.
Parametry
indeks wyjściowy |
---|
Rzuca
Wyjątek IllegalStateException | jeśli tłumacz jest zamknięty. |
---|
publiczne uruchomienie puste ( obiekt [] wejścia, mapa < liczba całkowita , obiekt > wyjścia)
Uruchamia wnioskowanie modelu na wielu danych wejściowych i zwraca wiele wyników.
Parametry
wejścia | tablica danych wejściowych. Dane wejściowe powinny być w tej samej kolejności, co dane wejściowe modelu. Każde dane wejściowe mogą być tablicą lub tablicą wielowymiarową albo ByteBuffer typów pierwotnych, w tym int, float, long i byte. ByteBuffer to preferowany sposób przekazywania dużych danych wejściowych, podczas gdy typy łańcuchowe wymagają użycia (wielowymiarowej) ścieżki wejściowej tablicy. Gdy używany jest ByteBuffer , jego zawartość powinna pozostać niezmieniona do czasu zakończenia wnioskowania o modelu. |
---|---|
wyjścia | mapa mapująca indeksy wyjściowe na wielowymiarowe tablice danych wyjściowych lub ByteBuffer typów pierwotnych, w tym int, float, long i byte. Musi jedynie przechowywać wpisy dotyczące wyjść, które mają być użyte. |