run.py

"""Script to compute analysis of T4 datasets."""

import argparse

from mmengine.logging import print_log

from tools.analysis_3d.analysis_runner import AnalysisRunner


def parse_args():
    """Add args and parse them through CLI."""
    parser = argparse.ArgumentParser(description="Create data info for T4dataset")
    parser.add_argument(
        "--config_path",
        type=str,
        required=True,
        help="config for T4dataset",
    )
    parser.add_argument(
        "--data_root_path",
        type=str,
        required=True,
        help="specify the root path of dataset",
    )
    parser.add_argument(
        "--dataset_version_config_root",
        type=str,
        required=True,
        help="specify the root path of dataset version config yaml files",
    )
    parser.add_argument(
        "-o",
        "--out_dir",
        type=str,
        required=True,
        help="output directory of info file",
    )
    args = parser.parse_args()
    return args


def main():
    """Main enrtypoint to run the Runner."""
    args = parse_args()
    # Build AnalysesRunner
    print_log("Building AnalysisRunner...", logger="current")
    analysis_runner = AnalysisRunner(
        data_root_path=args.data_root_path,
        dataset_version_config_root=args.dataset_version_config_root,
        config_path=args.config_path,
        out_path=args.out_dir,
    )
    print_log("Built AnalysisRunner!")

    # Run AnalysesRunner
    analysis_runner.run()


if __name__ == "__main__":
    main()