不知您是否能提供完整的测试样例呢 about tensorflowasr HOT 6 CLOSED

def stt(self,wav_path):

    am_result=self.am.predict(wav_path)
    if self.am.model_type=='Transducer':
        am_result =self.decode_am_result(am_result[1:-1])
        lm_result = self.lm.predict(am_result)
        lm_result = self.lm.decode(lm_result[0].numpy(), self.lm.word_featurizer)
    else:
        am_result=self.decode_am_result(am_result[0])
        lm_result=self.lm.predict(am_result)
        lm_result = self.lm.decode(lm_result[0].numpy(), self.lm.word_featurizer)
    return am_result,lm_result
def am_test(self,wav_path):
    #am_result is token id
    am_result = self.am.predict(wav_path)
    #token to vocab
    if self.am.model_type == 'Transducer':
        am_result = self.decode_am_result(am_result[1:-1])
    else:
        am_result = self.decode_am_result(am_result[0])
    return am_result
def lm_test(self,txt):
    py=pypinyin.pinyin(txt)
    input_py=[i[0] for i in py]
    #now lm_result is token id
    lm_result=self.lm.predict(input_py)
    #token to vocab
    lm_result=self.lm.decode(lm_result[0].numpy(),self.lm.word_featurizer)
    return lm_result