43 lines
1.4 KiB
JavaScript
43 lines
1.4 KiB
JavaScript
const sharp = require('sharp');
|
||
const Tesseract = require('tesseract.js');
|
||
const path = require('path');
|
||
|
||
/**
|
||
* 处理图像并进行验证码识别
|
||
* @param {string} inputImageName - 输入图像文件名（位于 .images 文件夹内）
|
||
* @param {string} outputImageName - 输出处理后图像的文件名（位于 .images 文件夹内）
|
||
* @returns {Promise<string>} - 返回识别的文本结果
|
||
*/
|
||
async function processAndRecognizeCaptcha(inputImagePath, outputImagePath) {
|
||
try {
|
||
// 图像处理
|
||
await sharp(inputImagePath)
|
||
// .sharpen()
|
||
.modulate({
|
||
brightness: 1.2, // 增加亮度
|
||
contrast: 1.5, // 增强对比度
|
||
})
|
||
.resize(800) // 调整图像宽度为800像素，保持纵横比
|
||
.grayscale() // 转换为灰度图
|
||
.threshold(128) // 二值化，阈值设定为128
|
||
.toFile(outputImagePath);
|
||
|
||
console.log('图像处理完成:', outputImagePath);
|
||
|
||
// 图像识别
|
||
const {
|
||
data: { text },
|
||
} = await Tesseract.recognize(outputImagePath, 'eng', {
|
||
langPath: './tessdata',
|
||
});
|
||
|
||
console.log('识别结果:', text.trim());
|
||
return text.replace(/\s+/g, '').trim(); // 返回识别的文本结果
|
||
} catch (err) {
|
||
console.error('处理或识别出错:', err);
|
||
throw err; // 将错误抛出以供调用者处理
|
||
}
|
||
}
|
||
|
||
module.exports = { processAndRecognizeCaptcha };
|