View source on GitHub |
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.