def main(): """ Main function to spawn the train and test process. """ args = parse_args() cfg = load_config(args) # Perform training. if cfg.TRAIN.ENABLE: launch_job(cfg=cfg, init_method=args.init_method, func=train) # Perform multi-clip testing. if cfg.TEST.ENABLE: launch_job(cfg=cfg, init_method=args.init_method, func=test) # Perform model visualization. if cfg.TENSORBOARD.ENABLE and ( cfg.TENSORBOARD.MODEL_VIS.ENABLE or cfg.TENSORBOARD.WRONG_PRED_VIS.ENABLE ): launch_job(cfg=cfg, init_method=args.init_method, func=visualize) # Run demo. if cfg.DEMO.ENABLE: demo(cfg)
def main(): """ Main function to spawn the train and test process. """ args = parse_args() cfg = load_config(args) #cfg.DEMO.WEBCAM = 0 cfg.DEMO.WEBCAM = -1 cfg.DEMO.INPUT_VIDEO = "demo_test/demo_in2.mp4" cfg.NUM_GPUS = 1 cfg.TRAIN.ENABLE = False cfg.TEST.ENABLE = False cfg.DEMO.OUTPUT_FILE = "demo_test/demo_out2.mp4" cfg.DEMO.ENABLE = True # Perform training. if cfg.TRAIN.ENABLE: launch_job(cfg=cfg, init_method=args.init_method, func=train) # Perform multi-clip testing. if cfg.TEST.ENABLE: launch_job(cfg=cfg, init_method=args.init_method, func=test) # Perform model visualization. if cfg.TENSORBOARD.ENABLE and cfg.TENSORBOARD.MODEL_VIS.ENABLE: launch_job(cfg=cfg, init_method=args.init_method, func=visualize) # Run demo. if cfg.DEMO.ENABLE: demo(cfg)