main.py

from core import YOLOTrainer
from configs import MyConfig, load_parser

import warnings
warnings.filterwarnings("ignore")


if __name__ == '__main__':
    config = MyConfig()

    config.init_dependent_config()

    ## If you want to use command-line arguments, please uncomment the following line
    # config = load_parser(config)

    trainer = YOLOTrainer(config)

    if config.is_testing:
        trainer.predict(config)
    else:    
        trainer.run(config)