Have a question? Connect with the community at the TensorFlow Forum Visit Forum

tflite_model_maker.object_detector.create

Loads data and train the model for object detection.

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.