Initialize repository

training/train_3d.py

@@ -0,0 +1,40 @@
+"""
+训练 3D 立体验证码识别模型 (ThreeDCNN)
+
+用法: python -m training.train_3d
+"""
+
+from config import (
+    THREED_CHARS,
+    IMAGE_SIZE,
+    SYNTHETIC_3D_DIR,
+    REAL_3D_DIR,
+)
+from generators.threed_gen import ThreeDCaptchaGenerator
+from models.threed_cnn import ThreeDCNN
+from training.train_utils import train_ctc_model
+
+
+def main():
+    img_h, img_w = IMAGE_SIZE["3d"]
+    model = ThreeDCNN(chars=THREED_CHARS, img_h=img_h, img_w=img_w)
+
+    print("=" * 60)
+    print("训练 3D 立体验证码识别模型 (ThreeDCNN)")
+    print(f"  字符集: {THREED_CHARS}  ({len(THREED_CHARS)} 字符)")
+    print(f"  输入尺寸: {img_h}×{img_w}")
+    print("=" * 60)
+
+    train_ctc_model(
+        model_name="threed",
+        model=model,
+        chars=THREED_CHARS,
+        synthetic_dir=SYNTHETIC_3D_DIR,
+        real_dir=REAL_3D_DIR,
+        generator_cls=ThreeDCaptchaGenerator,
+        config_key="threed",
+    )
+
+
+if __name__ == "__main__":
+    main()