I rearranged the repository structure, added a .gitignore
file and split the code into two scripts:
script.py
trains the network with the frozen VGG16 model it is called like this:python3 script.py model_name.hb5
wheremodel_name.hb5
is the file where the script will save the first modelfine_tuning.py
is used for tuning the previous model. It is called likepython3 script.py model_name.hb5
wheremodel_name.hb5
is the name of the model it will fine-tune. The final model is saved asmodel.hb5