ImageClassifier

lớp cuối cùng công khai ImageClassifier

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.
  • Tenor điểm đầu ra ( 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

Danh sách < 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.
Danh sách < 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 .
Danh sách < Phân loại >
phân loại (hình ảnh MlImage )
Thực hiện phân loại thực tế trên MlImage được cung cấp.
Danh sách < 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 .
Trình phân loại hình ảnh tĩnh
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.
Trình phân loại hình ảnh tĩnh
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 .
Trình phân loại hình ảnh tĩnh
createFromFile (Ngữ cảnh, String modelPath)
Tạo một phiên bản ImageClassifier từ ImageClassifier.ImageClassifierOptions mặc định.
Trình phân loại hình ảnh tĩnh
createFromFile ( File modelFile)
Tạo một phiên bản ImageClassifier từ ImageClassifier.ImageClassifierOptions mặc định.
Trình phân loại hình ảnh tĩnh
createFromFileAndOptions (Ngữ cảnh ngữ cảnh, String modelPath, ImageClassifier.ImageClassifierOptions tùy chọn)
Tạo một phiên bản ImageClassifier từ ImageClassifier.ImageClassifierOptions .
Trình phân loại hình ảnh tĩnh
createFromFileAndOptions (Các tùy chọn File modelFile, ImageClassifier.ImageClassifierOptions )
Tạo một phiên bản ImageClassifier .

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:

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:

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