mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-01-30 16:57:08 +09:00
parent 25cf8e9546
commit f1f66e40b1
3 changed files with 0 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp
View File

Binary file not shown.

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp
View File

Binary file not shown.

39
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py
View File

@ -798,44 +798,6 @@ def main():
test_sets = [f"test-clean_sampling"]
test_dl = [test_clean_dl]
for test_set, test_dl in zip(test_sets, test_dl):
results_dict = decode_dataset(
dl=test_dl,
params=params,
model=model,
sp=sp,
word_table=word_table,
decoding_graph=decoding_graph,
)
results = results_dict['greedy_search']
#jsons = open(f"{params.manifest_dir}/userlibri/{test_set}/{option}.jsonl", 'r').readlines()
#new_jsons = open(f"{params.manifest_dir}/userlibri/{test_set}/{option}_p.jsonl", 'w')
res_dict = {}
for res in results:
res_dict[res[0]] = ' '.join(res[2])
res_dict = sorted(res_dict.items(), key=lambda x:x[0])
for k, v in res_dict:
print(k, v)
exit()
if 0:
for line in jsons:
splited = line.split()
utt_id = splited[1][1:-2]
text_idx = splited.index('"text":')
pseudo = f'"greedy pseudo text": "{res_dict[utt_id]}",'
#splited.insert(text_idx, pseudo)
splited.insert(len(splited)-2, pseudo)
new_line = ' '.join(splited)
new_line += '\n'
new_jsons.write(new_line)
'''
for test_set, test_dl in zip(test_sets, test_dl):
results_dict = decode_dataset(
dl=test_dl,
@ -851,7 +813,6 @@ def main():
test_set_name=test_set,
results_dict=results_dict,
)
'''
'''
test_clean_cuts, test_clean_sets = librispeech.test_clean_cuts(option='user')