43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
|
import random
|
||
|
|
|
||
|
|
|
||
|
|
def sample_hotword(
|
||
|
|
length,
|
||
|
|
hotword_min_length,
|
||
|
|
hotword_max_length,
|
||
|
|
sample_rate,
|
||
|
|
double_rate,
|
||
|
|
pre_prob,
|
||
|
|
pre_index=None,
|
||
|
|
pre_hwlist=None,
|
||
|
|
):
|
||
|
|
if length < hotword_min_length:
|
||
|
|
return [-1]
|
||
|
|
if random.random() < sample_rate:
|
||
|
|
if pre_prob > 0 and random.random() < pre_prob and pre_index is not None:
|
||
|
|
return pre_index
|
||
|
|
if length == hotword_min_length:
|
||
|
|
return [0, length - 1]
|
||
|
|
elif random.random() < double_rate and length > hotword_max_length + hotword_min_length + 2:
|
||
|
|
# sample two hotwords in a sentence
|
||
|
|
_max_hw_length = min(hotword_max_length, length // 2)
|
||
|
|
# first hotword
|
||
|
|
start1 = random.randint(0, length // 3)
|
||
|
|
end1 = random.randint(start1 + hotword_min_length - 1, start1 + _max_hw_length - 1)
|
||
|
|
# second hotword
|
||
|
|
start2 = random.randint(end1 + 1, length - hotword_min_length)
|
||
|
|
end2 = random.randint(
|
||
|
|
min(length - 1, start2 + hotword_min_length - 1),
|
||
|
|
min(length - 1, start2 + hotword_max_length - 1),
|
||
|
|
)
|
||
|
|
return [start1, end1, start2, end2]
|
||
|
|
else: # single hotword
|
||
|
|
start = random.randint(0, length - hotword_min_length)
|
||
|
|
end = random.randint(
|
||
|
|
min(length - 1, start + hotword_min_length - 1),
|
||
|
|
min(length - 1, start + hotword_max_length - 1),
|
||
|
|
)
|
||
|
|
return [start, end]
|
||
|
|
else:
|
||
|
|
return [-1]
|