INNER CODE UNIT · Python

main

Nain57/Smart-AutoClicker · scripts/ocr-models/build_models.py:88

def main():
    dependency_checker.check_dependencies()

    parser = argparse.ArgumentParser(description="Full OCR Model Pipeline: Download -> ONNX -> NCNN")
    parser.add_argument("--output", required=True, help="Root output directory for models")
    parser.add_argument("--config", default=None, help="Path to models.yaml")
    parser.add_argument(
        "--mode",
        choices=["all", "default", "language_pack"],
        default="all",
        help="Build mode: 'default' (detection + latin), 'language_pack' (all others), or 'all' (default)"
    )

    args = parser.parse_args()

    output_root = os.path.abspath(args.output)
    yaml_path = args.config if args.config else os.path.join(os.path.dirname(__file__), "downloader", "models.yaml")

View source record →