Attention: This MediaPipe Solutions Preview is an early release. Learn more.

Module: mediapipe_model_maker.object_detector

MediaPipe Model Maker Python Public API For Object Detector.


class Dataset: Dataset library for object detector.

class HParams: The hyperparameters for training object detectors.

class ModelOptions: Configurable options for object detector model.

class ModelSpec: Specification of object detector model.

class ObjectDetector: ObjectDetector for building object detection model.

class ObjectDetectorOptions: Configurable options for building object detector.

class QATHParams: The hyperparameters for running quantization aware training (QAT) on object detectors.

class SupportedModels: Predefined object detector model specs supported by Model Maker.