Rust 解析验证码（结合 Tesseract OCR）

news/2025/11/15 23:52:39/文章来源:https://www.cnblogs.com/ocr12/p/19226556

环境准备
1.1 安装 Rust
如果尚未安装 Rust，可以使用官方推荐的 rustup 进行安装：

curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh

更多内容访问ttocr.com或联系1436423940
然后检查 Rust 版本：

rustc --version

1.2 安装 Tesseract OCR
macOS（Homebrew 安装）
brew install tesseract

Linux（Ubuntu 示例）
sudo apt update
sudo apt install tesseract-ocr -y

Windows

Windows 用户可以从 Tesseract 官方 GitHub
下载并安装 Tesseract，并配置环境变量。

检查 Tesseract 是否安装成功：

tesseract --version

1.3 创建 Rust 项目

创建一个新的 Rust 项目：

cargo new rust_ocr
cd rust_ocr

然后，在 Cargo.toml 中添加 Tesseract 相关依赖：

[dependencies]
tesseract = "0.13"
image = "0.24"

运行：

cargo build

代码实现
编辑 src/main.rs，写入以下代码：

use std::fs::File;
use std::path::Path;
use image::{GrayImage, Luma, DynamicImage, GenericImageView};
use tesseract::Tesseract;

fn preprocess_image(input_path: &str, output_path: &str) {
let img = image::open(input_path).expect("无法打开图片");

// 转换为灰度图像
let gray_img = img.into_luma8();

// 二值化处理
let mut binary_img = gray_img.clone();
for pixel in binary_img.pixels_mut() {
if pixel[0] > 128 {
*pixel = Luma([255]); // 白色
} else {
*pixel = Luma([0]); // 黑色
}
}

// 保存处理后的图像
binary_img.save(output_path).expect("无法保存处理后的图像");
}

fn recognize_captcha(image_path: &str) -> String {
let mut tess = Tesseract::new(None, "eng").expect("无法初始化 Tesseract");
tess.set_image(image_path);
tess.set_variable("tessedit_pageseg_mode", "6").unwrap(); // PSM 6 适用于验证码
tess.recognize().expect("OCR 识别失败");
tess.get_text().unwrap_or_else(|_| "识别失败".to_string())
}

fn main() {
let input_image = "captcha.png"; // 你的验证码图片路径
let processed_image = "processed_captcha.png";

// 预处理验证码图像
preprocess_image(input_image, processed_image);

// OCR 识别
let result = recognize_captcha(processed_image);
println!("识别出的验证码: {}", result);
}

代码解析
3.1 图像预处理
为了提高 OCR 识别率，我们对验证码进行了优化：

转换为灰度图像：

let gray_img = img.into_luma8();

二值化处理（增强对比度）：

for pixel in binary_img.pixels_mut() {
if pixel[0] > 128 {
*pixel = Luma([255]); // 白色
} else {
*pixel = Luma([0]); // 黑色
}
}

保存处理后的图像：

binary_img.save(output_path).expect("无法保存处理后的图像");

3.2 OCR 解析

初始化 Tesseract OCR：