Thực hiện tìm kiếm sự tương đồng 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ẽ. .
- 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
)-
N
thành phần tương ứng vớiN
kích thước của vectơ đặc tính được trả về cho lớp đầu ra này. - 2 hoặc 4 chiều, tức là
[1 x N]
hoặc[1 x 1 x 1 x N]
.
TODO(b/180502532): thêm con trỏ vào mô hình mẫu.
TODO(b/222671076): thêm các phương thức tạo nhà máy không có tùy chọn, chẳng hạn như `createFromFile`, sau khi hỗ trợ định dạng tệp đơn (tệp chỉ mục được đóng gói trong mô hình).
Các lớp lồng nhau
lớp học | ImageSearcher.ImageSearcherOptions | Các tùy chọn để thiết lập Trình tìm kiếm hình ảnh. |
Phương pháp công khai
Phương pháp kế thừa
Phương pháp công khai
ImageSearcher tĩnh công khai createFromBufferAndOptions (các tùy chọn ByteBuffer modelBuffer, ImageSearcher.ImageSearcherOptions )
Tạo một phiên bản ImageSearcher
với bộ đệm mô hình và ImageSearcher.ImageSearcherOptions
.
Thông số
bộ đệm mô hình | ByteBuffer trực tiếp hoặc MappedByteBuffer của mô hình tìm kiếm |
---|---|
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 |
---|---|
IOException | nếu xảy ra lỗi I/O khi tải tệp chỉ mục |
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 |
ImageSearcher tĩnh công khai createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, ImageSearcher.ImageSearcherOptions options, int indexFd)
Thông số
bộ đệm mô hình | |
---|---|
tùy chọn | |
chỉ mụcFd |
ImageSearcher tĩnh công khai createFromFileAndOptions (Ngữ cảnh ngữ cảnh, String modelPath, ImageSearcher.ImageSearcherOptions tùy chọn)
Tạo một phiên bản ImageSearcher
từ ImageSearcher.ImageSearcherOptions
.
Thông số
bối cảnh | |
---|---|
đường dẫn mô hình | đường dẫn của mô hình tìm kiếm có 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 hoặc tệp chỉ mục |
---|---|
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 |
ImageSearcher tĩnh công khai createFromFileAndOptions ( Tùy chọn File modelFile, ImageSearcher.ImageSearcherOptions )
Tạo một phiên bản ImageSearcher
.
Thông số
tập tin mô hình | mô hình tìm kiếm File 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 hoặc tệp chỉ mục |
---|---|
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 |
Danh sách công khai < NearestNeighbor > tìm kiếm (hình ảnh TensorImage , tùy chọn ImageProcessingOptions )
Thực hiện trích xuất nhúng trên TensorImage
được cung cấp với ImageProcessingOptions
, sau đó là tìm kiếm lân cận gần nhất trong chỉ mục.
ImageSearcher
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
.
ImageSearcher
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 < Hàng xóm gần nhất > tìm kiếm ( hình ảnh MlImage )
Thực hiện trích xuất nhúng trên MlImage
được cung cấp, sau đó là tìm kiếm lân cận gần nhất trong chỉ mục.
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 < Hàng xóm gần nhất > tìm kiếm ( hình ảnh MlImage , tùy chọn ImageProcessingOptions )
Thực hiện trích xuất nhúng trên MlImage
được cung cấp với ImageProcessingOptions
, sau đó là tìm kiếm lân cận gần nhất trong chỉ mục.
ImageSearcher
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ợ |
---|
Danh sách công khai < NearestNeighbor > tìm kiếm (hình ảnh TensorImage )
Thực hiện trích xuất nhúng trên TensorImage
được cung cấp, sau đó là tìm kiếm lân cận gần nhất trong chỉ mục.
ImageSearcher
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ợ |
---|