scripts/track.py

import os
import wandb

from omegaconf import OmegaConf
from pixel3dmm.tracking.tracker import Tracker
from pixel3dmm import env_paths

def main(cfg):
    tracker = Tracker(cfg)
    tracker.run()

if __name__ == '__main__':
    base_conf = OmegaConf.load(f'{env_paths.CODE_BASE}/configs/tracking.yaml')

    cli_conf = OmegaConf.from_cli()
    cfg = OmegaConf.merge(base_conf, cli_conf)

    #os.makedirs('/home/giebenhain/debug_wandb_p3dmm/', exist_ok=True)
    #wandb.init(
    #   dir='/home/giebenhain/debug_wandb_p3dmm/',
    #   #config=config,
    #   project='face-tracking-p3dmm',
    #   #tags=wandb_tags,
    #   #name=cfg.config_name,
#
    #)
    main(cfg)