问题描述
我正在尝试实施 deepspeech,作为实施的一部分,我正在尝试运行以下代码。我已经安装了 python3
和 deepspeech-0.9.3
在运行以下代码时,我收到一个错误 line 17,in <module> model = deepspeech.Model(MODEL_FILE_PATH,BEAM_WIDTH) TypeError: __init__() takes 2 positional arguments but 3 were given
import deepspeech
import numpy as np
import os
import pyaudio
import time
# DeepSpeech parameters
DEEPSPEECH_MODEL_DIR = 'deepspeech-0.9.3-models'
MODEL_FILE_PATH = os.path.join(DEEPSPEECH_MODEL_DIR,'output_graph.pbmm')
BEAM_WIDTH = 500
LM_FILE_PATH = os.path.join(DEEPSPEECH_MODEL_DIR,'lm.binary')
TRIE_FILE_PATH = os.path.join(DEEPSPEECH_MODEL_DIR,'trie')
LM_ALPHA = 0.75
LM_BETA = 1.85
# Make DeepSpeech Model
model = deepspeech.Model(MODEL_FILE_PATH,BEAM_WIDTH)
model.enableDecoderWithLM(LM_FILE_PATH,TRIE_FILE_PATH,LM_ALPHA,LM_BETA)
# Create a Streaming session
context = model.createStream()
# Encapsulate DeepSpeech audio Feeding into a callback for PyAudio
text_so_far = ''
def process_audio(in_data,frame_count,time_info,status):
global text_so_far
data16 = np.frombuffer(in_data,dtype=np.int16)
model.FeedAudioContent(context,data16)
text = model.intermediateDecode(context)
if text != text_so_far:
print('Interim text = {}'.format(text))
text_so_far = text
return (in_data,pyaudio.paContinue)
# PyAudio parameters
FORMAT = pyaudio.paInt16
CHANNELS = 1
RATE = 16000
CHUNK_SIZE = 1024
# Feed audio to deepspeech in a callback to PyAudio
audio = pyaudio.PyAudio()
stream = audio.open(
format=FORMAT,channels=CHANNELS,rate=RATE,input=True,frames_per_buffer=CHUNK_SIZE,stream_callback=process_audio
)
print('Please start speaking,when done press Ctrl-C ...')
stream.start_stream()
try:
while stream.is_active():
time.sleep(0.1)
except KeyboardInterrupt:
# PyAudio
stream.stop_stream()
stream.close()
audio.terminate()
print('Finished recording.')
# DeepSpeech
text = model.finishStream(context)
print('Final text = {}'.format(text))
解决方法
您要么提供了更多参数,要么没有在 init
中使用 self 关键字 ,您正在使用已弃用的代码。
您正在尝试使用 0.9 模型和 0.6 代码。从 0.7 开始,您不再使用 lm.binary 并尝试使用语言模型。而且你的代码还有很多错误。
也许使用 current examples. 之一,具体取决于您的用例是什么。