Hua
9b5f29083e
New solver subsystem with independent models: - GapDetectorCNN (1x128x256 grayscale → sigmoid) for slide gap detection - RotationRegressor (3x128x128 RGB → sin/cos via tanh) for rotation angle prediction - SlideSolver with 3-tier strategy: template match → edge detect → CNN fallback - RotateSolver with ONNX sin/cos → atan2 inference - Generators, training scripts, CLI commands, and slide track utility Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
28 lines
819 B
Python
28 lines
819 B
Python
"""
|
|
模型定义包
|
|
|
|
提供六种模型：
|
|
- CaptchaClassifier: 调度分类器 (轻量 CNN, < 500KB)
|
|
- LiteCRNN: 轻量 CRNN (普通字符 + 算式, < 2MB)
|
|
- ThreeDCNN: 3D 文字验证码专用模型 (ResNet-lite + BiLSTM, < 5MB)
|
|
- RegressionCNN: 回归 CNN (3D 旋转 + 滑块, ~1MB)
|
|
- GapDetectorCNN: 滑块缺口检测 CNN (~1MB)
|
|
- RotationRegressor: 旋转角度回归 sin/cos 编码 (~2MB)
|
|
"""
|
|
|
|
from models.classifier import CaptchaClassifier
|
|
from models.lite_crnn import LiteCRNN
|
|
from models.threed_cnn import ThreeDCNN
|
|
from models.regression_cnn import RegressionCNN
|
|
from models.gap_detector import GapDetectorCNN
|
|
from models.rotation_regressor import RotationRegressor
|
|
|
|
__all__ = [
|
|
"CaptchaClassifier",
|
|
"LiteCRNN",
|
|
"ThreeDCNN",
|
|
"RegressionCNN",
|
|
"GapDetectorCNN",
|
|
"RotationRegressor",
|
|
]
|