|
@@ -719,6 +719,7 @@ std::vector<std::vector<float>> Paraformer::CompileHotwordEmbedding(std::string
|
|
|
std::vector<int32_t> hotword_matrix;
|
|
std::vector<int32_t> hotword_matrix;
|
|
|
std::vector<int32_t> lengths;
|
|
std::vector<int32_t> lengths;
|
|
|
int hotword_size = 1;
|
|
int hotword_size = 1;
|
|
|
|
|
+ int real_hw_size = 0;
|
|
|
if (!hotwords.empty()) {
|
|
if (!hotwords.empty()) {
|
|
|
std::vector<std::string> hotword_array = split(hotwords, ' ');
|
|
std::vector<std::string> hotword_array = split(hotwords, ' ');
|
|
|
hotword_size = hotword_array.size() + 1;
|
|
hotword_size = hotword_array.size() + 1;
|
|
@@ -735,6 +736,9 @@ std::vector<std::vector<float>> Paraformer::CompileHotwordEmbedding(std::string
|
|
|
chars.insert(chars.end(), tokens.begin(), tokens.end());
|
|
chars.insert(chars.end(), tokens.begin(), tokens.end());
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
+ if(chars.size()==0){
|
|
|
|
|
+ continue;
|
|
|
|
|
+ }
|
|
|
std::vector<int32_t> hw_vector(max_hotword_len, 0);
|
|
std::vector<int32_t> hw_vector(max_hotword_len, 0);
|
|
|
int vector_len = std::min(max_hotword_len, (int)chars.size());
|
|
int vector_len = std::min(max_hotword_len, (int)chars.size());
|
|
|
for (int i=0; i<chars.size(); i++) {
|
|
for (int i=0; i<chars.size(); i++) {
|
|
@@ -743,8 +747,10 @@ std::vector<std::vector<float>> Paraformer::CompileHotwordEmbedding(std::string
|
|
|
}
|
|
}
|
|
|
std::cout << std::endl;
|
|
std::cout << std::endl;
|
|
|
lengths.push_back(vector_len);
|
|
lengths.push_back(vector_len);
|
|
|
|
|
+ real_hw_size += 1;
|
|
|
hotword_matrix.insert(hotword_matrix.end(), hw_vector.begin(), hw_vector.end());
|
|
hotword_matrix.insert(hotword_matrix.end(), hw_vector.begin(), hw_vector.end());
|
|
|
}
|
|
}
|
|
|
|
|
+ hotword_size = real_hw_size + 1;
|
|
|
}
|
|
}
|
|
|
std::vector<int32_t> blank_vec(max_hotword_len, 0);
|
|
std::vector<int32_t> blank_vec(max_hotword_len, 0);
|
|
|
blank_vec[0] = 1;
|
|
blank_vec[0] = 1;
|