27 lines
785 B
Python
27 lines
785 B
Python
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() |