Hua
788ddcae1a
- Add 57 unit tests covering generators, models, and pipeline components - Implement FastAPI HTTP service (server.py) with POST /solve and GET /health - Add checkpoint resume (断点续训) to both CTC and regression training utils - Fix device mismatch bug in CTC training (targets/input_lengths on GPU) - Add pytest dev dependency to pyproject.toml - Update .gitignore with data/solver/, data/real/, *.log - Remove PyCharm template main.py - Update training/__init__.py docs for solver training scripts Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
16 lines
832 B
Python
16 lines
832 B
Python
"""
|
|
训练脚本包
|
|
|
|
- dataset.py: CRNNDataset / CaptchaDataset / RegressionDataset 通用数据集类
|
|
- train_utils.py: CTC 训练通用逻辑 (train_ctc_model)
|
|
- train_regression_utils.py: 回归训练通用逻辑 (train_regression_model)
|
|
- train_normal.py: 训练普通字符识别 (LiteCRNN - normal)
|
|
- train_math.py: 训练算式识别 (LiteCRNN - math)
|
|
- train_3d_text.py: 训练 3D 立体文字识别 (ThreeDCNN)
|
|
- train_3d_rotate.py: 训练 3D 旋转回归 (RegressionCNN)
|
|
- train_3d_slider.py: 训练 3D 滑块回归 (RegressionCNN)
|
|
- train_classifier.py: 训练调度分类器 (CaptchaClassifier)
|
|
- train_slide.py: 训练滑块缺口检测 (GapDetectorCNN)
|
|
- train_rotate_solver.py: 训练旋转角度回归 (RotationRegressor)
|
|
"""
|