main.py

from src import constants, config, util
from train import train
from test import test
from preprocess import preprocess
from demo import demo


def main(sess_config, params):
    if params.help:
        print(params)
        exit(0)

    mode = params.mode.lower().strip()

    # If we are in modes other than download / pre-process load pre-existing configs.
    if mode in {constants.Modes.TRAIN, constants.Modes.TEST, constants.Modes.DEMO, constants.Modes.DEBUG}:
        params = util.load_config(params, util.config_path(params))  # Loads a pre-existing config otherwise == params

    if mode == constants.Modes.TRAIN:
        train(sess_config, params)
    elif mode == constants.Modes.DEBUG:
        train(sess_config, params, debug=True)
    elif mode == constants.Modes.TEST:
        test(sess_config, params)
    elif mode == constants.Modes.PREPROCESS:
        preprocess(params)
    elif mode == constants.Modes.DEMO:
        app = demo(sess_config, params)
        app.run(port=params.demo_server_port)
    else:
        print('Unknown Mode: {}'.format(mode))
        exit(0)


if __name__ == '__main__':
    defaults = util.namespace_json(path=constants.FilePaths.DEFAULTS)
    main(config.gpu_config(), config.model_config(defaults).FLAGS)