AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update max_pooling_RHE.py

Browse Source
This commit is contained in:
xiaohou 2021-11-18 15:54:24 +08:00 committed by GitHub
parent 24a0de00b8
commit 95ffddd981
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
kws/model/max_pooling_RHE.py
View File

@ -78,8 +78,10 @@ def max_pooling_RHE_binary_CE(logits,
max_ratio=1): max_ratio=1):
"""Max-pooling loss with regional hard example mining """Max-pooling loss with regional hard example mining
For each keyword utterance, select the frame with the highest posterior. For each keyword utterance, select the frame with the highest posterior.
The keyword is triggered when any of the frames is triggered. The keyword is triggered when any of the frames is triggered.
For each non-keyword utterance, select several hard examples using the RHE algorithm.
For each non-keyword utterance, select several hard examples using the
RHE algorithm.
Attributes: Attributes:
logits: (B, T, D), D is the number of keywords logits: (B, T, D), D is the number of keywords