o )j @s8ddlZddlZddlZGdddejjZddZdS)NcsBeZdZ d dejdejdeffdd Zdejfdd ZZS) GlobalCMVNTmeanistdnorm_varcs<t|j|jks J||_|d||d|dS)z Args: mean (torch.Tensor): mean stats istd (torch.Tensor): inverse std, std which is 1.0 / std rrN)super__init__shaperZregister_buffer)selfrrr __class__k/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/audio/kws/nearfield/cmvn.pyrs  zGlobalCMVN.__init__xcCs||j}|jr ||j}|S)z Args: x (torch.Tensor): (batch, max_len, feat_dim) Returns: (torch.Tensor): normalized feature )rrr)r rr r r forward(s  zGlobalCMVN.forward)T) __name__ __module__ __qualname__torchZTensorboolrr __classcell__r r r r rsrcCsd}d}t|}|}t|D]\}}|ddkrT|d}t|dks+J||d}td|d} | d} d d | D}t|t |dksSJq|d dkr|d}t|dksjJ||d}td|d} | d} d d | D}t|t |dksJq|d dkr|d}t|dksJ||d}td|d} | d}t|t |dt |dksJt|}qqWdn1swYt ||g}t |d|f}|S)z Load the kaldi format cmvn stats file and no need to calculate Args: cmvn_file: cmvn stats file in kaldi format Returns: a numpy array of [means, vars] NZAddShift z [\[](.*?)[\]]rcSsg|]}dt|qS)rfloat.0sr r r Ksz#load_kaldi_cmvn..ZRescalecSsg|]}t|qSr rrr r r rSsZSplice) open readlines enumeratefindstripsplitlenrefindallintnparrayZtile)Z cmvn_fileZmeansZvariancef all_linesidxlineZsegs next_lineZ means_strZ means_listZvars_strZ vars_listZ splice_strZ splice_listZ copy_timesZcmvnr r r load_kaldi_cmvn6sF     $ r2)r(numpyr+rnnModulerr2r r r r s