A股上市公司传智教育(股票代码 003032)旗下技术交流社区北京昌平校区

从 0 开始制作一个 NodeJS 命令行验证码识别工具。实现如下效果。


初始化项目# 创建 recognition 项目mkdir recognitioncd recognitionnpm init -y# 安装主依赖yarn add images tesseract.js# 安装工具依赖yarn add chalk yargs# 可选依赖yarn add socks5-http-client复制代码依赖说明
  • images:Node.js 轻量级跨平台图像编码库,用于处理下载下来的图片
  • tesseract.js:纯 JS 实现的 OCR(光学字符识别)工具,用于图像内容识别
  • chalk:让命令行内容样式好看
  • yargs:命令行参数解析器
  • socks5-http-client:SOCKS v5,用于设置代理,在需要拉取某些不能直接访问的资源时使用,request proxy 例子

项目准备新建 cli.js通常命令行工具入口名字为 cli.js,我们新建一个 cli.js 文件,并在开头写上:
#!/usr/bin/env node复制代码这样,我们告诉 *nix 系统,JavaScript 文件的解释器应该是 /usr/bin/env node,它查找本地安装的 node。
配置 bin// package.json{  "bin": {    "reg": "./cli.js"  }}复制代码这样配置完成后,别人 npm install -g @chenng/recognition 的包,就可以直接通过命令行运行了:
reg --url=https://static.chenng.cn/imgs/test_img.png复制代码link 本地开发我们如何能够在本地可以使用 rec 命令呢?只需要把本项目 link 即可:
yarn link复制代码核心逻辑主要逻辑在 cli.js 和 recognize.js 中。这里有几个注意点:
  • request 图片的时候要设置 encoding: null,否则返回的是乱码
  • 初次使用的时候需要下载训练集,需要花点时间
const Tesseract = require('tesseract.js');const images = require('images');const requset = require('request');const fs = require('fs');const { promisify } = require('util');const chalk = require('chalk');const writeFile = promisify(fs.writeFile);const rp = promisify(requset);class Recognize {  constructor(url) {    Recognize.downloadDir = `${__dirname}/dist/`;    Recognize.downloadFile = `${__dirname}/dist/temp.png`;    this.url = url;    this.start();  }  async start() {    const data = await this.downloadImg();    await writeFile(Recognize.downloadFile, data);    this.recognize();    const result = await Tesseract.recognize(Recognize.downloadFile, {      lang: 'eng',      tessedit_char_blacklist: 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ',    });    console.log(`      识别成功!      识别结果为:${chalk.green(result.text)}    `);  }  async downloadImg() {    if (!fs.existsSync(Recognize.downloadDir)) {      fs.mkdirSync(Recognize.downloadDir);      console.log(`创建了 ${Recognize.downloadDir} 文件夹`);    }        const res = await rp({      url: this.url,      method: 'GET',      encoding: null,    });    return res.body;  }  recognize() {    // 放大图片,并覆盖源文件    images(Recognize.downloadFile)      .size(400)      .save(Recognize.downloadFile);  }  }module.exports = Recognize;复制代码具体可以查看源码仓库:github.com/ringcrl/rec…
发布上线# 新建代码仓库,git push# 登录到 npmnpm adduser# 发包npm publish --access public# 全局安装npm install -g @chenng/recognition

【转载】仅作分享,侵删
作者:RingChenng
链接:https://juejin.im/post/5c0cc524e51d453cdd32498b



1 个回复

正序浏览
回复 使用道具 举报
您需要登录后才可以回帖 登录 | 加入黑马