scripts/infer_aris.py

import project_path
import argparse
from predict import predict_task
from datetime import datetime


def main(args):
    predict_task(args.aris, weights=args.weights)

def argument_parser():
    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)