File: phoneme_test.py

package info (click to toggle)
pocketsphinx 5.0.4-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 51,236 kB
  • sloc: ansic: 54,519; python: 2,438; sh: 566; cpp: 410; perl: 342; yacc: 93; lex: 50; makefile: 30
file content (31 lines) | stat: -rw-r--r-- 869 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#!/usr/bin/python

import os
import unittest
from pocketsphinx import Decoder, get_model_path

DATADIR = os.path.join(os.path.dirname(__file__), "../../test/data")


class PhonemeTest(unittest.TestCase):
    def test_phoneme(self):
        decoder = Decoder(
            allphone=get_model_path("en-us/en-us-phone.lm.bin"),
            lw=2.0, pip=0.3, beam=1e-200, pbeam=1e-20)

        # Decode streaming data.
        with open(os.path.join(DATADIR, "goforward.raw"), "rb") as stream:
            decoder.start_utt()
            while True:
                buf = stream.read(1024)
                if buf:
                    decoder.process_raw(buf, False, False)
                else:
                    break
            decoder.end_utt()

            print("Best phonemes: ", [seg.word for seg in decoder.seg()])


if __name__ == "__main__":
    unittest.main()