mp.tasks.vision.ImageClassifierOptions

Options for the image classifier task.

mp.tasks.vision.ImageClassifierOptions(
    base_options: mp.tasks.BaseOptions,
    running_mode: mp.tasks.vision.RunningMode = mp.tasks.vision.FaceDetectorOptions.running_mode,
    display_names_locale: Optional[str] = None,
    max_results: Optional[int] = None,
    score_threshold: Optional[float] = None,
    category_allowlist: Optional[List[str]] = None,
    category_denylist: Optional[List[str]] = None,
    result_callback: Optional[Callable[[ImageClassifierResult, image_module.Image, int], None]] = None
)

base_options Base options for the image classifier task.
running_mode The running mode of the task. Default to the image mode. Image classifier task has three running modes: 1) The image mode for classifying objects on single image inputs. 2) The video mode for classifying objects on the decoded frames of a video. 3) The live stream mode for classifying objects on a live stream of input data, such as from camera.
display_names_locale The locale to use for display names specified through the TFLite Model Metadata.
max_results The maximum number of top-scored classification results to return.
score_threshold Overrides the ones provided in the model metadata. Results below this value are rejected.
category_allowlist Allowlist of category names. If non-empty, classification results whose category name is not in this set will be filtered out. Duplicate or unknown category names are ignored. Mutually exclusive with category_denylist.
category_denylist Denylist of category names. If non-empty, classification results whose category name is in this set will be filtered out. Duplicate or unknown category names are ignored. Mutually exclusive with category_allowlist.
result_callback The user-defined result callback for processing live stream data. The result callback should only be specified when the running mode is set to the live stream mode.

Methods

__eq__

__eq__(
    other
)

category_allowlist None
category_denylist None
display_names_locale None
max_results None
result_callback None
running_mode <VisionTaskRunningMode.IMAGE: 'IMAGE'>
score_threshold None