Module: tflite_model_maker.object_detector

APIs to train an object detection model.

Classes

class DataLoader: DataLoader for object detector.

class EfficientDetSpec: A specification of the EfficientDet model.

class ObjectDetector: ObjectDetector class for inference and exporting to tflite.

Functions

EfficientDetLite0Spec(...): Creates EfficientDet-Lite0 model spec. See also: tflite_model_maker.object_detector.EfficientDetSpec.

EfficientDetLite1Spec(...): Creates EfficientDet-Lite1 model spec. See also: tflite_model_maker.object_detector.EfficientDetSpec.

EfficientDetLite2Spec(...): Creates EfficientDet-Lite2 model spec. See also: tflite_model_maker.object_detector.EfficientDetSpec.

EfficientDetLite3Spec(...): Creates EfficientDet-Lite3 model spec. See also: tflite_model_maker.object_detector.EfficientDetSpec.

EfficientDetLite4Spec(...): Creates EfficientDet-Lite4 model spec. See also: tflite_model_maker.object_detector.EfficientDetSpec.

create(...): Loads data and train the model for object detection.