-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathinference.py
More file actions
22 lines (21 loc) · 792 Bytes
/
Copy pathinference.py
File metadata and controls
22 lines (21 loc) · 792 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# To run
# python inference.py <audiofile> <number of time to run inference>
# python inference.py 8555-292519-0009.wav 3
import sys
import subprocess
import timeit
import shlex
# total arguments
n = len(sys.argv)
if n==1:
audiofile ='8555-292519-0009.wav'
run_times = 1
elif n==2:
audiofile = sys.argv[1]
run_times = 1
elif n==3:
audiofile = sys.argv[1]
run_times = sys.argv[2]
print('Running inference -',run_times,'time')
result = timeit.timeit('subprocess.run("deepspeech --model ./models/deepspeech-0.9.3-models.pbmm --scorer ./models/deepspeech-0.9.3-models.scorer --audio ./test_audios/"+audiofile, check=True, text=True, shell=True)',globals=globals(),number=int(run_times))
print('## Time taken for inference of',audiofile,str(run_times),'times -',result)