I am trying to fine-tune an existing pre-trained model for using ssd_object detection. I am taking inspiration from this notebook: models/eager_few_shot_od_training_tf2_colab.ipynb at master · tensorflow/models · GitHub
I am capable of saving my model after training, I get three files: checkpoint, ckpt-1.data-00000-of-00001 and ckpt-1.index
The model creation function procedure is:
Then I fine-tune the model and save it:
Assuming that I am doing things correctly when I want to re-initiate the model and load weights for inference, I am getting very bad results. Just like it’s not trained.
Restore weights function:
Can someone please help me find the problem?