scripts/aris_to_tracks.py

import argparse

import project_subpath
from backend.InferenceConfig import InferenceConfig
from backend.predict import predict_task


def main(args):
    predict_task(args.aris, config=InferenceConfig())

def argument_parser():
    """
    Run full inference on an aris file
    Args:
        aris (str): Path to ARIS file. Required.
        weights (str): Path to saved YOLOv5 weights. Default: ../models/v5m_896_300best.pt
    """
    parser = argparse.ArgumentParser()
    parser.add_argument("--aris", required=True, help="Path to ARIS file. Required.")
    parser.add_argument("--weights", default='../models/v5m_896_300best.pt', help="Path to saved YOLOv5 weights. Default: ../models/v5m_896_300best.pt")
    return parser

if __name__ == "__main__":
    args = argument_parser().parse_args()
    main(args)