Search code examples
tensorflowobject-detectionobject-detection-api

Can you export earlier checkpoints from the tensorflow object detection api v2?


I am training an object detection model up to 10 checkpoints using the tensorflow object detection api version 2. Exporting the final checkpoint using the exporter_main_v2.py works with no problems, however I would also like to export eg checkpoints 3, 6 and 8 to compare how they do in the actual setup. Is this possible?

I've tried deleting the later checkpoints and then running exporter_main_v2.py but this results in an error stating that there are later events in the events.out.tfevents file than the one I'm trying to export so it cant continue.


Solution

  • In your trained folder where checkpoints are, there is checkpoint key file, open that and change the "model_checkpoint_path" checkpoint-number in the first line. Generally it will be saved the last checkpoint.