INNER CODE UNIT · Python

main

Nain57/Smart-AutoClicker · scripts/ocr-models/converters/onnx_simplify.py:74

def main():
    """
    Main entry point for the ONNX simplifier script.
    """
    parser = argparse.ArgumentParser(description="Simplify an ONNX model using onnxsim.")

    parser.add_argument("--input", required=True, help="Input ONNX file")
    parser.add_argument("--output", required=True, help="Output simplified ONNX file")
    parser.add_argument("--shape", default="", help="Input shape e.g. 1,3,48,320")

    args = parser.parse_args()

    onnx_path = os.path.abspath(args.input)
    sim_path = os.path.abspath(args.output)

    print("\n=== ONNX Simplification ===")
    print("[INFO] Input:", onnx_path)
    print("[INFO] Output:", sim_path)

View source record →