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.
- wprowadzanie obrazu o rozmiarze
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
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:
- Region zainteresowania (ROI) (poprzez
ImageProcessingOptions.Builder.setRoi(Rect)
). Domyślnie jest to cały obraz. - obrót obrazu (poprzez
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). Domyślnie jest toImageProcessingOptions.Orientation.TOP_LEFT
.
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:
- Region zainteresowania (ROI) (poprzez
ImageProcessingOptions.Builder.setRoi(Rect)
). Domyślnie jest to cały obraz. - obrót obrazu (poprzez
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). Domyślnie jest toImageProcessingOptions.Orientation.TOP_LEFT
.MlImage.getRotation()
nie jest skuteczna.
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 )
Tworzy instancję ImageClassifier
z 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 |