INNER CODE UNIT · Python

main

Nain57/Smart-AutoClicker · scripts/ocr-models/converters/onnx_to_ncnn.py:90

def main():
    """
    Main entry point for the ONNX to NCNN converter script.
    """
    parser = argparse.ArgumentParser(description="Convert an ONNX model to NCNN using PNNX.")

    parser.add_argument("--input", required=True, help="Input ONNX file")
    parser.add_argument("--output", required=True, help="Output folder for NCNN model")
    parser.add_argument("--shape", default="", help="Input shape e.g. 1,3,48,320")
    parser.add_argument("--fp16", action="store_true", help="Enable FP16 mode")

    args = parser.parse_args()

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

    print("\n=== ONNX → NCNN Conversion ===")
    print("[INFO] Input:", onnx_path)

View source record →