|
|
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() |
