Model

Model klasy publicznej

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
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 MappedByteBuffer getData ()

Zwraca dane modelu odwzorowanego w pamięci.

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.

publiczny ciąg getPath ()

Zwraca ścieżkę pliku modelu przechowywanego w Assets.

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.