Thực hiện phân loại trên hình ảnh.
API yêu cầu một mô hình TFLite với Siêu dữ liệu mô hình TFLite tùy chọn nhưng được khuyến khích mạnh mẽ. .
API hỗ trợ các mô hình có một tenxơ đầu vào hình ảnh và một tenxơ đầu ra phân loại. Để cụ thể hơn, đây là các yêu cầu.
- Tenor hình ảnh đầu vào (
kTfLiteUInt8
/kTfLiteFloat32
)- hình ảnh đầu vào có kích thước
[batch x height x width x channels]
. - suy luận hàng loạt không được hỗ trợ (
batch
buộc phải là 1). - chỉ hỗ trợ đầu vào RGB (
channels
bắt buộc phải là 3). - nếu loại là
kTfLiteFloat32
thì bắt buộc phải đính kèm NormalizationOptions vào siêu dữ liệu để chuẩn hóa đầu vào.
- hình ảnh đầu vào có kích thước
kTfLiteUInt8
/ kTfLiteFloat32
)- với
N
lớp có 2 hoặc 4 chiều, chẳng hạn như[1 x N]
hoặc[1 x 1 x 1 x N]
- tệp nhãn bắt buộc phải được đóng gói vào siêu dữ liệu. Xem ví dụ về tạo siêu dữ liệu cho bộ phân loại hình ảnh . Nếu không có tệp nhãn nào được đóng gói, nó sẽ sử dụng chỉ mục làm nhãn trong kết quả.
Bạn có thể tìm thấy một ví dụ về mô hình như vậy trên TensorFlow Hub. .
Các lớp lồng nhau
lớp học | ImageClassifier.ImageClassifierOptions | Các tùy chọn để thiết lập ImageClassifier. |
Phương pháp công khai
Phương pháp kế thừa
Phương pháp công khai
Danh sách công khai < Phân loại > phân loại ( hình ảnh TensorImage )
Thực hiện phân loại thực tế trên TensorImage
được cung cấp.
ImageClassifier
hỗ trợ các loại không gian màu TensorImage
sau:
Thông số
hình ảnh | một đối tượng UINT8 TensorImage đại diện cho hình ảnh RGB hoặc YUV |
---|
Ném
Ngoại lệ Đối số bất hợp pháp | nếu loại hình ảnh không gian màu không được hỗ trợ |
---|
Danh sách công khai < Phân loại > phân loại ( hình ảnh TensorImage , tùy chọn ImageProcessingOptions )
Thực hiện phân loại thực tế trên TensorImage
được cung cấp với ImageProcessingOptions
.
ImageClassifier
hỗ trợ các tùy chọn sau:
- Vùng quan tâm (ROI) (thông qua
ImageProcessingOptions.Builder.setRoi(Rect)
). Nó mặc định cho toàn bộ hình ảnh. - xoay hình ảnh (thông qua
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). Nó mặc định làImageProcessingOptions.Orientation.TOP_LEFT
.
ImageClassifier
hỗ trợ các loại không gian màu TensorImage
sau:
Thông số
hình ảnh | một đối tượng UINT8 TensorImage đại diện cho hình ảnh RGB hoặc YUV |
---|---|
tùy chọn |
Ném
Ngoại lệ Đối số bất hợp pháp | nếu loại hình ảnh không gian màu không được hỗ trợ |
---|
Danh sách công khai < Phân loại > phân loại ( MlImage image)
Thực hiện phân loại thực tế trên MlImage
được cung cấp.
Thông số
hình ảnh | một đối tượng MlImage đại diện cho một hình ảnh |
---|
Ném
Ngoại lệ Đối số bất hợp pháp | nếu loại lưu trữ hoặc định dạng của hình ảnh không được hỗ trợ |
---|
Danh sách công khai < Phân loại > phân loại ( hình ảnh MlImage , tùy chọn ImageProcessingOptions )
Thực hiện phân loại thực tế trên MlImage
được cung cấp với ImageProcessingOptions
.
ImageClassifier
hỗ trợ các tùy chọn sau:
- Vùng quan tâm (ROI) (thông qua
ImageProcessingOptions.Builder.setRoi(Rect)
). Nó mặc định cho toàn bộ hình ảnh. - xoay hình ảnh (thông qua
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). Nó mặc định làImageProcessingOptions.Orientation.TOP_LEFT
.MlImage.getRotation()
không hiệu quả.
Thông số
hình ảnh | một đối tượng MlImage đại diện cho một hình ảnh |
---|---|
tùy chọn | định cấu hình các tùy chọn bao gồm ROI và xoay vòng |
Ném
Ngoại lệ Đối số bất hợp pháp | nếu loại lưu trữ hoặc định dạng của hình ảnh không được hỗ trợ |
---|
ImageClassifier tĩnh công khai createFromBuffer ( ByteBuffer modelBuffer)
Tạo một phiên bản ImageClassifier
với bộ đệm mô hình và ImageClassifier.ImageClassifierOptions
mặc định.
Thông số
bộ đệm mô hình | ByteBuffer trực tiếp hoặc MappedByteBuffer của mô hình phân loại |
---|
Ném
Ngoại lệ Đối số bất hợp pháp | nếu bộ đệm mô hình không phải là ByteBuffer trực tiếp hoặc MappedByteBuffer |
---|---|
Ngoại lệ nhà nước bất hợp pháp | nếu có lỗi nội bộ |
Ngoại lệ thời gian chạy | nếu có một lỗi không xác định khác |
ImageClassifier tĩnh công khai createFromBufferAndOptions (các tùy chọn ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions )
Tạo một phiên bản ImageClassifier
với bộ đệm mô hình và ImageClassifier.ImageClassifierOptions
.
Thông số
bộ đệm mô hình | ByteBuffer trực tiếp hoặc MappedByteBuffer của mô hình phân loại |
---|---|
tùy chọn |
Ném
Ngoại lệ Đối số bất hợp pháp | nếu bộ đệm mô hình không phải là ByteBuffer trực tiếp hoặc MappedByteBuffer |
---|---|
Ngoại lệ nhà nước bất hợp pháp | nếu có lỗi nội bộ |
Ngoại lệ thời gian chạy | nếu có một lỗi không xác định khác |
ImageClassifier tĩnh công khai createFromFile (Ngữ cảnh bối cảnh, String modelPath)
Tạo một phiên bản ImageClassifier
từ ImageClassifier.ImageClassifierOptions
mặc định.
Thông số
bối cảnh | |
---|---|
đường dẫn mô hình | đường dẫn của mô hình phân loại với siêu dữ liệu trong nội dung |
Ném
IOException | nếu xảy ra lỗi I/O khi tải mô hình tflite |
---|---|
Ngoại lệ Đối số bất hợp pháp | nếu một đối số không hợp lệ |
Ngoại lệ nhà nước bất hợp pháp | nếu có lỗi nội bộ |
Ngoại lệ thời gian chạy | nếu có một lỗi không xác định khác |
ImageClassifier tĩnh công khai createFromFile ( File modelFile)
Tạo một phiên bản ImageClassifier
từ ImageClassifier.ImageClassifierOptions
mặc định.
Thông số
tập tin mô hình | mô hình File loại Ví dụ tệp |
---|
Ném
IOException | nếu xảy ra lỗi I/O khi tải mô hình tflite |
---|---|
Ngoại lệ Đối số bất hợp pháp | nếu một đối số không hợp lệ |
Ngoại lệ nhà nước bất hợp pháp | nếu có lỗi nội bộ |
Ngoại lệ thời gian chạy | nếu có một lỗi không xác định khác |
ImageClassifier tĩnh công khai createFromFileAndOptions (Ngữ cảnh bối cảnh, String modelPath, ImageClassifier.ImageClassifierOptions tùy chọn)
Tạo một phiên bản ImageClassifier
từ ImageClassifier.ImageClassifierOptions
.
Thông số
bối cảnh | |
---|---|
đường dẫn mô hình | đường dẫn của mô hình phân loại với siêu dữ liệu trong nội dung |
tùy chọn |
Ném
IOException | nếu xảy ra lỗi I/O khi tải mô hình tflite |
---|---|
Ngoại lệ Đối số bất hợp pháp | nếu một đối số không hợp lệ |
Ngoại lệ nhà nước bất hợp pháp | nếu có lỗi nội bộ |
Ngoại lệ thời gian chạy | nếu có một lỗi không xác định khác |
ImageClassifier tĩnh công khai createFromFileAndOptions ( Tùy chọn File modelFile, ImageClassifier.ImageClassifierOptions )
Tạo một phiên bản ImageClassifier
.
Thông số
tập tin mô hình | mô hình File loại Ví dụ tệp |
---|---|
tùy chọn |
Ném
IOException | nếu xảy ra lỗi I/O khi tải mô hình tflite |
---|---|
Ngoại lệ Đối số bất hợp pháp | nếu một đối số không hợp lệ |
Ngoại lệ nhà nước bất hợp pháp | nếu có lỗi nội bộ |
Ngoại lệ thời gian chạy | nếu có một lỗi không xác định khác |