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)
