-
Notifications
You must be signed in to change notification settings - Fork 10
/
test.py
94 lines (68 loc) · 3.66 KB
/
test.py
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
from Notq.Notq import *
# Microsoft
subscription="<paste-your-speech-key-here>"
region="<paste-your-speech-location/region-here>"
sampleRate = 4000
# os.getcwd() returns path of current directory
singlePath_voice_mp3 = os.getcwd()+"\\VOICE_AD"
filename = "titleh.mp3.wav"
directory_voice = os.getcwd()+"\\VOICE_AD"
directory_text = os.getcwd()+"\\text"
directory_resample = os.getcwd()+"\\VOICE_AD"
singlePath_resample = os.getcwd()+"\\VOICE_AD\\_12341.mp3.wav"
singlePath_convert = os.getcwd()+"\\VOICE_AD\\titleh.mp3"
sentimentModelPath = 'HooshvareLab/bert-fa-base-uncased'
similarityModelPath = os.getcwd()+"\\cc.fa.300.bin"
## Test /convert_dir_mp3_to_wav/ (singleFilePath == False)
''' convert_dir_mp3_to_wav(directory_voice , singleFilePath = False) '''
## Test /convert_dir_mp3_to_wav/ (singleFilePath == True)
''' convert_dir_mp3_to_wav(singlePath_convert , singleFilePath = True) '''
## Test /resample/ (singleFilePath == False)
''' resample(directory_resample , sampleRate , False) '''
## Test /resample/ (singleFilePath == True)
''' resample(singlePath_resample , sampleRate , True) '''
## Test /speechToText/ (VOSK_wav)
''' speechToText("VOSK_wav" , filename , directory_voice , directory_text) '''
## Test /speechToText/ (Google_wav)
''' speechToText("Google_wav" , filename , directory_voice , directory_text) '''
## Test /speechToText/ (Microsoft)
''' speechToText("Microsoft" , filename , directory_voice , directory_text, subscription, region) '''
## Test /speechToText/ (default)
''' speechToText("",filename , directory_voice , directory_text) '''
## Test /silenceTime/
''' print(silenceTime(singlePath_resample)) '''
## TEST /similarity/
''' from numpy.linalg import norm
def cosine_sim(a, b):
return np.inner(a, b) / (norm(a) * norm(b))
m_model = similarity("c:/Users/Shaghayegh/Desktop/new_stt/Notq/cc.fa.300.bin")
x = m_model.get_sentence_vector('دو نفر در بیمارستان بستری بودند .یک نفر میتوانست تکون بخوره اما دیگری نمیتونست')
y = m_model.get_sentence_vector('دو نفر تو بیمارستان بستری بودن.یکیشون میتونست تکون بخوره اما دیگری نه')
z = m_model.get_sentence_vector('دو تا مرد بستری بودند که یکیشون میتونست حرکت بکنه اما اون یکی نه')
w = m_model.get_sentence_vector('یک')
xy = cosine_sim(x,y)
yz = cosine_sim(y,z)
zx = cosine_sim(x,z)
xw = cosine_sim(x,w)
wy = cosine_sim(y,w)
wz = cosine_sim(z,w)
print(xy,yz,zx,xw,wy,wz) '''
## TEST /sentiment/
''' print(sentiment(['دوست داشتنی بود'
,'واقعا از قدرت نویسنده لذت بردم داستان کوتاهیه ولی به قول یکی از دوستان داستان تا همیشه گوشه‌ای از ذهن ادم‌میمونه خیلی خوب بود خیلی'
,'اصلا خوب نبود'])) '''
## Test /caclulate_fluency/ (defualt)
''' print(caclulate_fluency(singlePath_resample)) '''
## Test /caclulate_fluency/ (SpeechRate)
''' print(caclulate_fluency(singlePath_resample , "SpeechRate")) '''
## Test /caclulate_fluency/ (ArticulationRate)
''' print(caclulate_fluency(singlePath_resample , "ArticulationRate")) '''
## Test /caclulate_fluency/ (PhonationTimeRatio)
''' print(caclulate_fluency(singlePath_resample , "PhonationTimeRatio")) '''
## Test /caclulate_fluency/ (MeanLengthOfRuns)
''' print(caclulate_fluency(singlePath_resample , "MeanLengthOfRuns")) '''
## Test /split_wavfile/
''' split_wavfile(singlePath_resample , directory_voice) '''
## Test /emotion Recognition/
''' path = "/path/to/your_file.wav"
outputs = emotionRecognition(path) '''