30分鐘實現小程式語音識別
為了參加某個作秀活動,研究了一波如何結合小程式、科大訊飛實現語音錄入、識別的實現。科大訊飛開發文件中只給出 Python 的 demo,並沒有給出 node.js 的 sdk,但問題不大。本文將從小程式相關程式碼到最後對接科大訊飛 api 過程,一步步介紹,半個小時,搭建完成小程式語音識別功能!不能再多了!當然,前提是最好掌握有一點點小程式、node.js 甚至是音訊相關的知識。
架構先行
架構比較簡單,大夥兒可以先看下圖。除了小程式,需要提供 3 個服務,檔案上傳、音訊編碼及對接科大訊飛的服務。
node.js 對接科大訊飛的 api,npm 上已經有同學提供了 sdk,有興趣的同學可以去搜索瞭解一下,筆者這裡是直接呼叫了科大訊飛的 api 介面。

擼起袖子加油幹
1、建立小程式
鵝廠的小程式文件非常詳細,在這裡筆者就不對如何建立一個小程式的步驟進行詳細闡述了。有需要的同學可以檢視鵝廠的小程式開發文件。
1.1 相關程式碼
我們摘取小程式裡面,語音錄入和語音上傳部分的程式碼。
// 根據wx提供的api建立錄音管理物件 const recorderManager = wx.getRecorderManager(); // 監聽語音識別結束後的行為 recorderManager.onStop(recorderResponse => { // tempFilePath 是錄製的音訊檔案 const { tempFilePath } = recorderResponse; // 上傳音訊檔案,完成語音識別翻譯 wx.uploadFile({ url: 'http://127.0.0.1:7001/voice', // 該服務在後面搭建。另外,小程式釋出時要求後臺服務提供https服務!這裡的地址僅為開發環境配置。 filePath: tempFilePath, name: 'file', complete: res => { console.log(res); // 我們期待res,就是翻譯後的內容 } }); }); // 開始錄音,觸發條件可以是按鈕或其他,由你自己決定 recorderManager.start({ duration: 5000 // 最長錄製時間 // 其他引數可以預設,更多引數可以檢視https://developers.weixin.qq.com/miniprogram/dev/api/media/recorder/RecorderManager.start.html }); 複製程式碼
2、搭建檔案伺服器
步驟 1 程式碼中提到了一個 url 地址大家應該都還記得。
http://127.0.0.1:7001/voice 複製程式碼
小程式本身還並沒有提供語音識別的功能,所以在這裡我們需要藉助於“後端”服務的能力,完成我們語音識別翻譯的功能。
2.1 egg.js 服務初始化
我們使用 egg.js 的 cli 快速初始化一個工程,當然你也可以使用 express、koa、kraken 等等框架,框架的選型在此不是重點我們就不做展開闡述了。對 egg.js 不熟悉的同學可以檢視egg.js 的官網。
npm i egg-init -g egg-init voice-server --type=simple cd voice-server npm i 複製程式碼
安裝完成後,執行以下程式碼
npm run dev 複製程式碼
隨後訪問瀏覽器http://127.0.0.1:7001應該可以看到一個Hi, egg 的頁面。至此我們的服務初始化完成。
2.2 檔案上傳介面
a) 修改 egg.js 的檔案上傳配置
開啟 config/config.default.js,新增以下兩項配置
module.exports = appInfo => { ... config.multipart = { fileSize: '2gb', // 限制檔案大小 whitelist: [ '.aac', '.m4a', '.mp3' ], // 支援上傳的檔案字尾名 }; config.security = { csrf: { enable: false // 關閉csrf } }; ... } 複製程式碼
b) 新增 VoiceController
開啟 app/controller 資料夾,新建檔案 voice.js。編寫 VoiceController 使其繼承於 egg.js 的 Controller。具體程式碼如下:
const Controller = require('egg').Controller; const fs = require('fs'); const path = require('path'); const pump = require('mz-modules/pump'); const uuidv1 = require('uuid/v1'); // 依賴於uuid庫,用於生成唯一檔名,使用npm i uuid安裝即可 // 音訊檔案上傳後儲存的路徑 const targetPath = path.resolve(__dirname, '..', '..', 'uploads'); class VoiceController extends Controller { constructor(params) { super(params); if (!fs.existsSync(targetPath)) { fs.mkdirSync(targetPath); } } async translate() { const parts = this.ctx.multipart({ autoFields: true }); let stream; const voicePath = path.join(targetPath, uuidv1()); while (!isEmpty((stream = await parts()))) { await pump(stream, fs.createWriteStream(voicePath)); } // 到這裡就完成了檔案上傳。如果你不需要檔案落地,也可以在後續的操作中,直接使用stream操作檔案流 ... // 音訊編碼 // 科大訊飛語音識別 ... } } 複製程式碼
c) 最後一步,新增路由規則
寫完 controller 之後,我們依據 egg.js 的規則,在 router.js 裡面新增一個路由。
module.exports = app => { const { router, controller } = app; router.get('/', controller.home.index); router.get('/voice', controller.voice.translate); }; 複製程式碼
OK,至此你可以測試一下從小程序錄音,錄音完成後上傳到後臺檔案伺服器的完整流程。如果沒問題,那恭喜你你已經完成了 80%的工作了!
3、音訊編碼服務
在上文中,小程序錄音的方法 recorderManager.start 的時候我們提及到了“更多引數”。其中有一個引數是 format,支援 aac 和 mp3 兩種(預設是 aac)。然後我們查閱了科大訊飛的 api 文件,音訊編碼支援“未壓縮的 pcm 或 wav 格式”。
什麼 aac、pcm、wav?emmm.. OK,我們只是前端,既然格式不對等,那隻需要完成 aac -> pcm 轉化即可,ffmpeg 立即浮現在筆者的腦海裡。一番搜尋,命令大概是這樣子的:
ffmpeg -i uploads/a3f588d0-edf8-11e8-b6f5-2929aef1b7f8.aac -f s16le -ar 8000 -ac 2 -y decoded.pcm # -i 後面帶的是原始檔 # -f s16le 指的是編碼格式 # -ar 8000 編碼位元速率 # -ac 2 通道 複製程式碼
接下來我們使用 node.js 來實現上述命令。
3.1 引入相關依賴包
npm i ffmpeg-static npm i fluent-ffmpeg 複製程式碼
3.2 建立一個編碼服務
在 app/service 資料夾中,建立 ffmpeg.js 檔案。新建 FFmpegService 繼承於 egg.js 的 Service
const { Service } = require('egg'); const ffmpeg = require('fluent-ffmpeg'); const ffmpegStatic = require('ffmpeg-static'); const path = require('path'); const fs = require('fs'); ffmpeg.setFfmpegPath(ffmpegStatic.path); class FFmpegService extends Service { async aac2pcm(voicePath) { const command = ffmpeg(voicePath); // 方便測試,我們將轉碼後文件落地到磁碟 const targetDir = path.join(path.dirname(voicePath), 'pcm'); if (!fs.existsSync(targetDir)) { fs.mkdirSync(targetDir); } const target = path.join(targetDir, path.basename(voicePath)) + '.pcm'; return new Promise((resolve, reject) => { command .audioCodec('pcm_s16le') .audioChannels(2) .audioBitrate(8000) .output(target) .on('error', error => { reject(error); }) .on('end', () => { resolve(target); }) .run(); }); } } module.exports = FFmpegService; 複製程式碼
3.3 呼叫 ffmpegService,獲得 pcm 檔案
回到 app/controller/voice.js 檔案中,我們在檔案上傳完成後,呼叫 ffmpegService 提供的 aac2pcm 方法,獲取到 pcm 檔案的路徑。
// app/controller/voice.js ... async translate() { ... ... const pcmPath = await this.ctx.service.ffmpeg.aac2pcm(voicePath); ... } ... 複製程式碼
4、對接科大訊飛 API
首先,需要到科大訊飛開放平臺註冊並新增應用、開通應用的語音聽寫服務。
我們再寫一個服務,在 app/service 資料夾下建立 xfyun.js 檔案,實現 XFYunService 繼承於 egg.js 的 Service。
4.1 引入相關依賴
npm i axios // 網路請求庫 npm i md5 // 科大訊飛介面中需要md5計算 npm i form-urlencoded // 介面中需要對部分內容進行urlencoded 複製程式碼
4.2 XFYunService 實現
const { Service } = require('egg'); const fs = require('fs'); const formUrlencoded = require('form-urlencoded').default; const axios = require('axios'); const md5 = require('md5'); const API_KEY = 'xxxx'; // 在科大訊飛控制檯上可以查到服務的APIKey const API_ID = 'xxxxx'; // 同樣可以在控制檯查到 class XFYunService extends Service { async voiceTranslate(voicePath) { // 繼上文,暴力的讀取檔案 let data = fs.readFileSync(voicePath); // 將內容進行base64編碼 data = new Buffer(data).toString('base64'); // 進行url encode data = formUrlencoded({ audio: data }); const params = { engine_type: 'sms16k', aue: 'raw' }; const x_CurTime = Math.floor(new Date().getTime() / 1000) + '', x_Param = new Buffer(JSON.stringify(params)).toString('base64'); return axios({ url: 'http://api.xfyun.cn/v1/service/v1/iat', method: 'POST', data, headers: { 'X-Appid': API_ID, 'X-CurTime': x_CurTime, 'X-Param': x_Param, 'X-CheckSum': md5(API_KEY + x_CurTime + x_Param) } }).then(res => { // 查詢成功後,返回response的data return res.data || {}; }); } } module.exports = XFYunService; 複製程式碼
4.3 呼叫 XFYunService,完成語音識別
再次回到 app/controller/voice.js 檔案中,我們在 ffmpeg 轉碼完成後,呼叫 XFYunService 提供的 voiceTranslate 方法,完成語音識別。
// app/controller/voice.js ... async translate() { ... ... const result = await this.ctx.service.xfyun.voiceTranslate(pcmPath); this.ctx.body = result; if (+result.code !== 0) { this.ctx.status = 500; } } ... 複製程式碼
至此我們完成語音識別的程式碼編寫。主要流程其實很簡單,通過小程序錄入語音檔案,上傳到檔案伺服器之後,通過 ffmpeg 獲取到 pcm 檔案, 最後再轉發到科大訊飛的 api 介面進行識別。
以上,如有錯漏,歡迎指正!
@Author: _Jay