Honk is a PyTorch reimplementation of Google's TensorFlow convolutional neural networks for keyword spotting, which accompanies the recent release of their Speech Commands Dataset. For more details, please consult our writeup:
- Raphael Tang, Jimmy Lin. Honk: A PyTorch Reimplementation of Convolutional Neural Networks for Keyword Spotting. arXiv:1710.06554, October 2017.
Honk is useful for building on-device speech recognition capabilities for interactive intelligent agents. Our code can be used to identify simple commands (e.g., "stop" and "go") and be adapted to detect custom "command triggers" (e.g., "Hey Siri!").
Check out this video for a demo of Honk in action!
Use the instructions below to run the demo application (shown in the above video) yourself!
Currently, PyTorch has official support for only Linux and OS X. Thus, Windows users will not be able to run this demo easily.
To deploy the demo, run the following commands:
- If you do not have PyTorch, please see the website.
- Install Python dependencies:
pip install -r requirements.txt
- Install GLUT (OpenGL Utility Toolkit) through your package manager (e.g.
apt-get install freeglut3-dev
) - Fetch the data and models:
./fetch_data.sh
- Start the PyTorch server:
python .
- Run the demo:
python utils/speech_demo.py
If you need to adjust options, like turning off CUDA, please edit config.json
.
Additional notes for Mac OS X:
- GLUT is already installed on Mac OS X, so that step isn't needed.
- If you have issues installing pyaudio, this may be the issue.
python .
deploys the web service for identifying if audio contain the command word. By default, config.json
is used for configuration, but that can be changed with --config=<file_name>
.
If the server is behind a firewall, one workflow is to create an SSH tunnel and use port forwarding with the port specified in config (default 16888).
POST /listen
Args (JSON):
wav_data
: 16kHz sampling rate, 16-bit PCM mono-channel raw audio data (with no WAVE header), gzipped and base64-encoded.
Returns (JSON):
contains_command
:true
ifwav_data
contains the command word,false
otherwise.
For a real-time example, please see utils/client.py
.
Unfortunately, the QA client has no support for the general public yet, since it requires a custom QA service. However, it can still be used to retarget the command keyword.
python client.py
runs the QA client. You may retarget a keyword by doing python client.py --mode=retarget
. Please note that text-to-speech may not work well on Linux distros; in this case, please supply IBM Watson credentials via --watson-username
and --watson--password
. You can view all the options by doing python client.py -h
.
python model.py --mode [train|eval]
trains or evaluates the model. It expects all training examples to follow the same format as that of Speech Commands Dataset. The recommended workflow is to download the dataset and add custom keywords, since the dataset already contains many useful audio samples and background noise.
There are command options available:
option | input format | default | description |
---|---|---|---|
--batch_size |
[1, n) | 100 | the mini-batch size to use |
--cache_size |
[0, inf) | 32768 | number of items in audio cache, consumes around 32 KB * n |
--conv1_pool |
[1, inf) [1, inf) | 2 2 | the width and height of the pool filter |
--conv1_size |
[1, inf) [1, inf) | 10 4 | the width and height of the conv filter |
--conv1_stride |
[1, inf) [1, inf) | 1 1 | the width and length of the stride |
--conv2_pool |
[1, inf) [1, inf) | 1 1 | the width and height of the pool filter |
--conv2_size |
[1, inf) [1, inf) | 10 4 | the width and height of the conv filter |
--conv2_stride |
[1, inf) [1, inf) | 1 1 | the width and length of the stride |
--data_folder |
string | /data/speech_dataset | path to data |
--dev_every |
[1, inf) | 10 | dev interval in terms of epochs |
--dev_pct |
[0, 100] | 10 | percentage of total set to use for dev |
--dropout_prob |
[0.0, 1.0) | 0.5 | the dropout rate to use |
--gpu_no |
[-1, n] | 1 | the gpu to use |
--group_speakers_by_id |
{true, false} | true | whether to group speakers across train/dev/test |
--input_file |
string | the path to the model to load | |
--input_length |
[1, inf) | 16000 | the length of the audio |
--lr |
(0.0, inf) | 0.001 | the learning rate to use |
--mode |
{train, eval} | train | the mode to use |
--model |
string | cnn-trad-pool2 | one of cnn-trad-pool2 , cnn-tstride-{2,4,8} , cnn-tpool{2,3} , cnn-one-fpool3 , cnn-one-fstride{4,8} |
--n_dct_filters |
[1, inf) | 40 | the number of DCT bases to use |
--n_epochs |
[0, inf) | 500 | number of epochs |
--n_feature_maps1 |
[1, inf) | 64 | the number of feature maps for conv net 1 |
--n_feature_maps2 |
[1, inf) | 64 | the number of feature maps for conv net 2 |
--n_labels |
[1, n) | 4 | the number of labels to use |
--n_mels |
[1, inf) | 40 | the number of Mel filters to use |
--no_cuda |
switch | false | whether to use CUDA |
--noise_prob |
[0.0, 1.0] | 0.8 | the probability of mixing with noise |
--output_file |
string | model/google-speech-dataset.pt | the file to save the model to |
--seed |
(inf, inf) | 0 | the seed to use |
--silence_prob |
[0.0, 1.0] | 0.1 | the probability of picking silence |
--test_pct |
[0, 100] | 10 | percentage of total set to use for testing |
--timeshift_ms |
[0, inf) | 100 | time in milliseconds to shift the audio randomly |
--train_pct |
[0, 100] | 80 | percentage of total set to use for training |
--unknown_prob |
[0.0, 1.0] | 0.01 | the probability of picking an unknown word |
--wanted_words |
string1 string2 ... stringn | command random | the desired target words |
You may do the following to record sequential audio and save to the same format as that of speech command dataset:
python manage_audio.py record
Input any key (return is fastest) to open the microphone. After one second of silence, recording automatically halts.
Several options are available:
--output-prefix: Prefix of the output audio sequence
--min-sound-lvl: Minimum sound level at which audio is not considered silent
--timeout-seconds: Duration of silence after which recording halts
--output-begin-index: Starting sequence number
python manage_audio.py listen
This assists in setting sane values for --min-sound-lvl
for recording.
python manage_audio.py generate-contrastive --directory [directory]
generates contrastive examples from all .wav files in [directory]
using phonetic segmentation.
Speech command dataset contains one-second-long snippets of audio.
python manage_audio.py trim --directory [directory]
trims to the loudest one-second for all .wav files in [directory]
. The careful user should manually check all audio samples using an audio editor like Audacity.