Support external FunCaptcha ONNX fallback

2026-03-12 19:53:51 +08:00
parent 221af67bf4
commit 201e517b86
7 changed files with 181 additions and 13 deletions
--- a/inference/export_onnx.py
+++ b/inference/export_onnx.py
@@ -205,6 +205,7 @@ def _load_and_export(model_name: str):
        metadata = {
            "model_name": model_name,
            "task": "funcaptcha_siamese",
+            "preprocess": "rgb_centered",
            "question": question,
            "num_candidates": int(ckpt.get("num_candidates", task_cfg["num_candidates"])),
            "tile_size": list(ckpt.get("tile_size", task_cfg["tile_size"])),