moge/scripts/cli.py

import os
os.environ['OPENCV_IO_ENABLE_OPENEXR'] = '1'
from pathlib import Path
import sys
if (_package_root := str(Path(__file__).absolute().parents[2])) not in sys.path:
    sys.path.insert(0, _package_root)

import click


@click.group(help='MoGe command line interface.')
def cli():
    pass

def main():
    from moge.scripts import app, infer, infer_baseline, infer_panorama, eval_baseline, vis_data
    cli.add_command(app.main, name='app')
    cli.add_command(infer.main, name='infer')
    cli.add_command(infer_baseline.main, name='infer_baseline')
    cli.add_command(infer_panorama.main, name='infer_panorama')
    cli.add_command(eval_baseline.main, name='eval_baseline')
    cli.add_command(vis_data.main, name='vis_data')
    cli()


if __name__ == '__main__':
    main()