main.py

from data_loader.reader import FruitsDataLoader
from models.template_model import TemplateModel
from trainers.fruit_model_trainer import FruitModelTrainer
from utils.config import process_config
from utils.dirs import create_dirs
from utils.utils import get_args

def main():
    # capture the config path from the run arguments
    # then process the json configuration file
    try:
        args = get_args()
        config = process_config(args.config)
    except:
        print("missing or invalid arguments")
        exit(0)

    # create the experiments dirs
    create_dirs([config.callbacks.tensorboard_log_dir, config.callbacks.checkpoint_dir])

    print('Create the data generator.')
    data_loader = FruitsDataLoader(config)

    print('Create the model.')
    model = TemplateModel(config)

    print('Create the trainer')

    trainer = FruitModelTrainer(model.model, data_loader.get_train_val_data(), config)

    print('Start training the model.')
    trainer.train()


if __name__ == '__main__':
    main()