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
Args |
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.
|
Returns |
An instance based on ObjectDetector.
|