Module: tflite_support.task.vision

View source on GitHub

TensorFlow Lite Task Library Vision APIs.

This module provides interface to run TensorFlow Lite computer vision models.

Classes

class ImageClassifier: Class that performs classification on images.

class ImageClassifierOptions: Options for the image classifier task.

class ImageEmbedder: Class that performs dense feature vector extraction on images.

class ImageEmbedderOptions: Options for the image embedder task.

class ImageSearcher: Class to performs image search.

class ImageSearcherOptions: Options for the image search task.

class ImageSegmenter: Class that performs segmentation on images.

class ImageSegmenterOptions: Options for the image segmenter task.

class ObjectDetector: Class that performs object detection on images.

class ObjectDetectorOptions: Options for the object detector task.

class TensorImage: Wrapper class for the Image object.