Align task API and add FunCaptcha support

2026-03-12 19:32:59 +08:00
parent ef9518deeb
commit bc6776979e
33 changed files with 3446 additions and 672 deletions
--- a/inference/init.py
+++ b/inference/init.py
@@ -2,16 +2,19 @@
 推理包

 - pipeline.py:    CaptchaPipeline 核心推理流水线
+- fun_captcha.py: FunCaptcha 专项推理
 - export_onnx.py: PyTorch → ONNX 导出
 - math_eval.py:   算式计算模块
 """

 from inference.pipeline import CaptchaPipeline
+from inference.fun_captcha import FunCaptchaRollballPipeline
 from inference.math_eval import eval_captcha_math
 from inference.export_onnx import export_model, export_all

 __all__ = [
    "CaptchaPipeline",
+    "FunCaptchaRollballPipeline",
    "eval_captcha_math",
    "export_model",
    "export_all",