ImageClassifier

publiczna klasa końcowa ImageClassifier

Wykonuje klasyfikację obrazów.

Interfejs API oczekuje modelu TFLite z opcjonalnymi, ale zdecydowanie zalecanymi metadanymi modelu TFLite. .

Interfejs API obsługuje modele z jednym tensorem wejściowym obrazu i jednym tensorem wyjściowym klasyfikacji. Mówiąc dokładniej, oto wymagania.

  • Tensor obrazu wejściowego ( kTfLiteUInt8 / kTfLiteFloat32 )
    • wprowadzanie obrazu o rozmiarze [batch x height x width x channels] .
    • wnioskowanie wsadowe nie jest obsługiwane (wymagana jest batch 1).
    • obsługiwane są tylko wejścia RGB (wymagane są 3 channels ).
    • jeśli typ to kTfLiteFloat32 , do metadanych należy dołączyć NormalizationOptions w celu normalizacji danych wejściowych.
  • Tensor wyniku wyjściowego ( kTfLiteUInt8 / kTfLiteFloat32 )
    • z N klasami o 2 lub 4 wymiarach, takimi jak [1 x N] lub [1 x 1 x 1 x N]
    • plik etykiety musi zostać spakowany do metadanych. Zobacz przykład tworzenia metadanych dla klasyfikatora obrazu . Jeśli nie spakowano żadnych plików etykiet, w wyniku użyje indeksu jako etykiety.

Przykład takiego modelu można znaleźć na TensorFlow Hub. .

Klasy zagnieżdżone

klasa ImageClassifier.ImageClassifierOptions Opcje konfigurowania ImageClassifier.

Metody publiczne

Lista <Klasyfikacje>
klasyfikuj (obraz TensorImage )
Wykonuje rzeczywistą klasyfikację na podanym TensorImage .
Lista <Klasyfikacje>
klasyfikuj (obraz TensorImage , opcje ImageProcessingOptions )
Wykonuje rzeczywistą klasyfikację na podanym TensorImage z ImageProcessingOptions .
Lista <Klasyfikacje>
klasyfikuj ( obraz MlImage )
Wykonuje rzeczywistą klasyfikację na podanym MlImage .
Lista <Klasyfikacje>
klasyfikuj (obraz MlImage , opcje ImageProcessingOptions )
Wykonuje rzeczywistą klasyfikację na podanym MlImage za pomocą ImageProcessingOptions .
statyczny klasyfikator obrazu
createFromBuffer ( ByteBuffer modelBuffer)
Tworzy instancję ImageClassifier z buforem modelu i domyślną ImageClassifier.ImageClassifierOptions .
statyczny klasyfikator obrazu
statyczny klasyfikator obrazu
createFromFile (kontekst kontekstowy, String modelPath)
Tworzy wystąpienie ImageClassifier z domyślnego ImageClassifier.ImageClassifierOptions .
statyczny klasyfikator obrazu
createFromFile ( plik modelFile)
Tworzy wystąpienie ImageClassifier z domyślnego ImageClassifier.ImageClassifierOptions .
statyczny klasyfikator obrazu
statyczny klasyfikator obrazu

Metody dziedziczone

Metody publiczne

lista publiczna <Klasyfikacje> klasyfikuj (obraz TensorImage )

Wykonuje rzeczywistą klasyfikację na podanym TensorImage .

ImageClassifier obsługuje następujące typy przestrzeni kolorów TensorImage :

Parametry
obraz obiekt TensorImage UINT8, który reprezentuje obraz RGB lub YUV
Rzuca
Wyjątek IllegalArgument jeśli typ przestrzeni kolorów obrazu nie jest obsługiwany

lista publiczna <Klasyfikacje> klasyfikuj (obraz TensorImage , opcje ImageProcessingOptions )

Wykonuje rzeczywistą klasyfikację na podanym TensorImage z ImageProcessingOptions .

ImageClassifier obsługuje następujące opcje:

ImageClassifier obsługuje następujące typy przestrzeni kolorów TensorImage :

Parametry
obraz obiekt TensorImage UINT8, który reprezentuje obraz RGB lub YUV
opcje
Rzuca
Wyjątek IllegalArgument jeśli typ przestrzeni kolorów obrazu nie jest obsługiwany

lista publiczna <Klasyfikacje> klasyfikuj (obraz MlImage )

Wykonuje rzeczywistą klasyfikację na podanym MlImage .

Parametry
obraz obiekt MlImage reprezentujący obraz
Rzuca
Wyjątek IllegalArgument jeśli typ przechowywania lub format obrazu nie jest obsługiwany

lista publiczna <Klasyfikacje> klasyfikuj (obraz MlImage , opcje ImageProcessingOptions )

Wykonuje rzeczywistą klasyfikację na podanym MlImage za pomocą ImageProcessingOptions .

ImageClassifier obsługuje następujące opcje:

Parametry
obraz obiekt MlImage reprezentujący obraz
opcje konfiguruje opcje, w tym ROI i rotację
Rzuca
Wyjątek IllegalArgument jeśli typ przechowywania lub format obrazu nie jest obsługiwany

publiczny statyczny ImageClassifier createFromBuffer ( ByteBuffer modelBuffer)

Tworzy instancję ImageClassifier z buforem modelu i domyślną ImageClassifier.ImageClassifierOptions .

Parametry
Bufor modelu bezpośredni ByteBuffer lub MappedByteBuffer modelu klasyfikacji
Rzuca
Wyjątek IllegalArgument jeśli bufor modelu nie jest bezpośrednim ByteBuffer lub MappedByteBuffer
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

publiczny statyczny ImageClassifier createFromBufferAndOptions (opcje ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions )

Tworzy instancję ImageClassifier z buforem modelu i ImageClassifier.ImageClassifierOptions .

Parametry
Bufor modelu bezpośredni ByteBuffer lub MappedByteBuffer modelu klasyfikacji
opcje
Rzuca
Wyjątek IllegalArgument jeśli bufor modelu nie jest bezpośrednim ByteBuffer lub MappedByteBuffer
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

public static ImageClassifier createFromFile (kontekst kontekstowy, string modelPath)

Tworzy wystąpienie ImageClassifier z domyślnego ImageClassifier.ImageClassifierOptions .

Parametry
kontekst
Ścieżka modelu ścieżka modelu klasyfikacji z metadanymi w zasobach
Rzuca
Wyjątek IO jeśli wystąpi błąd we/wy podczas ładowania modelu tflite
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

publiczny statyczny ImageClassifier createFromFile ( plik modelFile)

Tworzy wystąpienie ImageClassifier z domyślnego ImageClassifier.ImageClassifierOptions .

Parametry
plik modelu model klasyfikacji Instancja File
Rzuca
Wyjątek IO jeśli wystąpi błąd we/wy podczas ładowania modelu tflite
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

public static ImageClassifier createFroFileAndOptions (kontekst kontekstowy, String modelPath, opcje ImageClassifier.ImageClassifierOptions )

Parametry
kontekst
Ścieżka modelu ścieżka modelu klasyfikacji z metadanymi w zasobach
opcje
Rzuca
Wyjątek IO jeśli wystąpi błąd we/wy podczas ładowania modelu tflite
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

public static ImageClassifier createFromFileAndOptions ( Opcje pliku modelFile, ImageClassifier.ImageClassifierOptions )

Tworzy instancję ImageClassifier .

Parametry
plik modelu model klasyfikacji Instancja File
opcje
Rzuca
Wyjątek IO jeśli wystąpi błąd we/wy podczas ładowania modelu tflite
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd