mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
32 lines
1.5 KiB
Python
32 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):
|
|
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] |