tflite_model_maker.object_detector.create

Stay organized with collections Save and categorize content based on your preferences.

Loads data and train the model for object detection.

@classmethod
tflite_model_maker.object_detector.create(
    train_data: tflite_model_maker.object_detector.DataLoader,
    model_spec: tflite_model_maker.object_detector.EfficientDetSpec,
    validation_data: Optional[tflite_model_maker.object_detector.DataLoader] = None,
    epochs: Optional[tflite_model_maker.object_detector.DataLoader] = None,
    batch_size: Optional[int] = None,
    train_whole_model: bool = False,
    do_train: bool = True
) -> T

Used in the notebooks

Used in the tutorials

train_data Training data.
model_spec Specification for the model.
validation_data Validation data. If None, skips validation process.
epochs Number of epochs for training.
batch_size Batch size for training.
train_whole_model Boolean, False by default. If true, train the whole model. Otherwise, only train the layers that are not match model_spec.config.var_freeze_expr.
do_train Whether to run training.

An instance based on ObjectDetector.