File size: 785 Bytes
201ab98 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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() |