CODEWITHSUNDEEP
tensorflowmachine-learninggoogle-cloud-mlobject-detection-api
Simon Labrecque
Simon Labrecque

Reputation: 597

How to continue training an object detection model using Tensorflow Object Detection API?

I'm using Tensorflow Object Detection API to train an object detection model using transfer learning. Specifically, I'm using ssd_mobilenet_v1_fpn_coco from the model zoo, and using the sample pipeline provided, having of course replaced the placeholders with actual links to my training and eval tfrecords and labels.

I was able able to successfully train a model on my ~5000 images (and corresponding bounding boxes) using the above pipeline (I'm mainly using Google's ML Engine on TPU, if revelant).

Now, I prepared an additional ~2000 images, and would like continue training my model with those new images, without restarting from scratch (training the initial model took ~6h of TPU time). How can I do that?

Upvotes: 7

Views: 5211

Answers (2)

netanel-sam
netanel-sam

Reputation: 1912

You have two options, in both you need to change the input_path of the train_input_reader of your new dataset:

  1. When specifying a checkpoint to fine-tune in the training configuration, specify the checkpoint of your trained model
train_config{
    fine_tune_checkpoint: <path_to_your_checkpoint>
    fine_tune_checkpoint_type: "detection"
    load_all_detection_checkpoint_vars: true
}
  1. Simply keep using the same configuration (except the train_input_reader) with the same model_dir of your previous model. That way, the API will create a graph and will check whether a checkpoint already exists in model_dir and fits the graph. If so - it will restore it and continue training it.

Edit: fine_tune_checkpoint_type was previously set as true by mistake, while it should be "detection" or "classification" in general, and "detection" in this specific case. Thanks Krish for noticing.

Upvotes: 8

Vlad-HC
Vlad-HC

Reputation: 4757

I haven't retrained the object detection model on a new dataset, but it looks like increasing the number of training steps train_config.num_steps in the config file and also adding images in the tfrecord files should be enough for that.

Upvotes: 1

Related Questions