当前位置：首页 > news >正文

用 Swift 结合 Tesseract OCR 进行验证码识别

news 2026/6/29 13:48:10

环境准备
1.1 安装 Xcode 和 Swift

确保你的 macOS 设备上安装了 Xcode：

xcode-select --install

检查 Swift 版本：

swift --version

1.2 安装 Tesseract OCR
更多内容访问ttocr.com或联系1436423940
使用 Homebrew 安装 Tesseract：

brew install tesseract

安装完成后，检查版本：

tesseract --version

1.3 创建 Swift 项目

创建一个新的 Swift 项目：

mkdir SwiftOCR
cd SwiftOCR
swift package init --type executable

1.4 添加依赖

在 Package.swift 文件中添加 SwiftTesseract 依赖：

// swift-tools-version:5.7
import PackageDescription

let package = Package(
name: "SwiftOCR",
dependencies: [
.package(url: "https://github.com/gali8/Tesseract-OCR-iOS.git", from: "5.4.0")
],
targets: [
.executableTarget(
name: "SwiftOCR",
dependencies: ["Tesseract-OCR-iOS"]
)
]
)

然后运行：

swift build

代码实现

在 Sources/SwiftOCR/main.swift 文件中编写如下代码：

import Foundation
import TesseractOCR
import AppKit

// 预处理图像
func preprocessImage(imagePath: String) -> NSImage? {
guard let image = NSImage(contentsOfFile: imagePath) else {
print("无法加载图像")
return nil
}

// 将图像转换为灰度
let grayImage = imageToGrayScale(image: image)// 保存处理后的图像（可选）
let outputPath = "processed_captcha.png"
saveImage(image: grayImage, path: outputPath)return grayImage

}

// 灰度化处理
func imageToGrayScale(image: NSImage) -> NSImage {
let ciImage = CIImage(data: image.tiffRepresentation!)
let filter = CIFilter(name: "CIColorControls")!
filter.setValue(ciImage, forKey: kCIInputImageKey)
filter.setValue(0.0, forKey: kCIInputSaturationKey) // 设为灰度
let outputCIImage = filter.outputImage!

let rep = NSCIImageRep(ciImage: outputCIImage)
let nsImage = NSImage(size: rep.size)
nsImage.addRepresentation(rep)return nsImage

}

// 保存图像
func saveImage(image: NSImage, path: String) {
guard let tiffData = image.tiffRepresentation,
let bitmap = NSBitmapImageRep(data: tiffData),
let pngData = bitmap.representation(using: .png, properties: [:]) else {
return
}
try? pngData.write(to: URL(fileURLWithPath: path))
}

// OCR 识别
func recognizeCaptcha(imagePath: String) {
guard let processedImage = preprocessImage(imagePath: imagePath) else {
return
}

if let tesseract = G8Tesseract(language: "eng") {tesseract.image = processedImagetesseract.recognize()if let recognizedText = tesseract.recognizedText {print("识别出的验证码: \(recognizedText.trimmingCharacters(in: .whitespacesAndNewlines))")} else {print("OCR 识别失败")}
} else {print("无法初始化 Tesseract")
}

}

// 运行程序
let imagePath = "captcha.png" // 确保该路径下有验证码图片
recognizeCaptcha(imagePath: imagePath)