captcha/README.md

Captcha Recognition

基于深度神经网络(DNN)的验证码识别

使用方法

1. 克隆项目到本地

git clone https://git.taurusxin.com/taurusxin/captcha.git
cd captcha

2. 创建虚拟环境并安装依赖

python -m venv .venv

# Windows
.venv\Scripts\Activate.ps1

# Linux/MacOS
source .venv/bin/activate

# 先安装 PyTorch GPU 版本，cuda 12.4
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

# 然后再安装其他依赖
pip install -r requirements.txt

3. 根据提示生成数据集，生成3次数据集，分别用于训练，用于测试，用于验证。

python captcha_gen.py

建议的数据集长度如下：

数据集	长度
Train	50000
Test	1000
Predict	30

4. 训练模型

python train.py

5. 测试模型

python test.py

6. 预测验证码

python predict.py

7. 输出 onnx 模型

python export_onnx.py