\;j+jddlZddlmZmZddlZddlmZ ddeeefdedeeeffdZ ddeeefdedeeeffd Z dd e d ed eded e def dZ dde de d e defdZ dde de d e d ed eededee efd e defdZ ddedededeedef dZ d de d e dee d e def dZy)!N)OptionalUnion)Tensorfreqhtkreturnc|rLt|trdtjd|dz zzSdt jd|dz zzSd}d}||z |z }d}||z |z }t j ddz }t|trO|tj ||z d z|z z}||kDj |j} || z|d | z zz}|S||k\r!|t j ||z d z|z z}|S) aConvert Hz to Mels. Args: freq (Union[Tensor, float]): The input tensor with arbitrary shape. htk (bool, optional): Use htk scaling. Defaults to False. Returns: Union[Tensor, float]: Frequency in mels. Examples: .. code-block:: python >>> import paddle >>> val = 3.0 >>> htk_flag = True >>> mel_paddle_tensor = paddle.audio.functional.hz_to_mel( ... paddle.to_tensor(val), htk_flag) F@?@竪P@@@皙@;@绽|=) isinstancerpaddlelog10mathlogastypedtype) rrf_minf_spmels min_log_hz min_log_mellogsteptargetmasks kG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/audio/functional/functional.py hz_to_melr$s+. dF #FLLte|);<< <DJJsTE\'9:: : E D 5LD DJ%-Khhsmd"G$ &**TJ%6%>?'I I z!))$**5}t H   K : $*;e*C!Dw!NND Kmelc|rdd|dz zdz zSd}d}|||zz}d}||z |z }tjdd z }t|trL|t j |||z zz}||kDj |j} || z|d | z zz}|S||k\r|tj |||z zz}|S) a0Convert mel bin numbers to frequencies. Args: mel (Union[float, Tensor]): The mel frequency represented as a tensor with arbitrary shape. htk (bool, optional): Use htk scaling. Defaults to False. Returns: Union[float, Tensor]: Frequencies in Hz. Examples: .. code-block:: python >>> import paddle >>> val = 3.0 >>> htk_flag = True >>> mel_paddle_tensor = paddle.audio.functional.mel_to_hz( ... paddle.to_tensor(val), htk_flag) ... r $@r r r rrrrr)rrrrrexprr) r&rrrfreqsrrr r!r"s r# mel_to_hzr+Ns. v.455 E D D3J EJ%-Khhsmd"G#vfjjC+4E)FGGk!))#))4  H!   L + 'S;5F*G!HHE Lr%n_melsrf_maxrct||}t||}tj||||}t||}|S)aNCompute mel frequencies. Args: n_mels (int, optional): Number of mel bins. Defaults to 64. f_min (float, optional): Minimum frequency in Hz. Defaults to 0.0. fmax (float, optional): Maximum frequency in Hz. Defaults to 11025.0. htk (bool, optional): Use htk scaling. Defaults to False. dtype (str, optional): The data type of the return frequencies. Defaults to 'float32'. Returns: Tensor: Tensor of n_mels frequencies in Hz with shape `(n_mels,)`. Examples: .. code-block:: python >>> import paddle >>> n_mels = 64 >>> f_min = 0.5 >>> f_max = 10000 >>> htk_flag = True >>> paddle_mel_freq = paddle.audio.functional.mel_frequencies( ... n_mels, f_min, f_max, htk_flag, 'float64') )rr)r$rlinspacer+) r,rr-rrmin_melmax_melrr*s r#mel_frequenciesr3{sBB3'G3'G ??7GV5 AD d $E Lr%srn_fftc jtjdt|dz td|dzz|S)aCompute fourier frequencies. Args: sr (int): Sample rate. n_fft (int): Number of fft bins. dtype (str, optional): The data type of the return frequencies. Defaults to 'float32'. Returns: Tensor: FFT frequencies in Hz with shape `(n_fft//2 + 1,)`. Examples: .. code-block:: python >>> import paddle >>> sr = 16000 >>> n_fft = 128 >>> fft_freq = paddle.audio.functional.fft_frequencies(sr, n_fft) rrr/)rr0floatintr4r5rs r#fft_frequenciesr;s.( ??1eBi!mSUaZ-@ NNr%normc|t|dz }tj|td|dzzf|}t |||} t |dz||||} | dd| ddz } | j d| j dz } t|D]_} | |  | | z }| | dz| | dzz }tjtj|tj|||| <a|d k(r'd | d|dz| d|z z }||j dz}|St|ttfr,tjjj||d }|S) apCompute fbank matrix. Args: sr (int): Sample rate. n_fft (int): Number of fft bins. n_mels (int, optional): Number of mel bins. Defaults to 64. f_min (float, optional): Minimum frequency in Hz. Defaults to 0.0. f_max (Optional[float], optional): Maximum frequency in Hz. Defaults to None. htk (bool, optional): Use htk scaling. Defaults to False. norm (Union[str, float], optional): Type of normalization. Defaults to 'slaney'. dtype (str, optional): The data type of the return matrix. Defaults to 'float32'. Returns: Tensor: Mel transform matrix with shape `(n_mels, n_fft//2 + 1)`. Examples: .. code-block:: python >>> import paddle >>> sr = 23 >>> n_fft = 51 >>> fbank = paddle.audio.functional.compute_fbank_matrix(sr, n_fft) Nr7rr/r:)rr-rrrslaney@)paxis)r8rzerosr9r;r3 unsqueezerangemaximum zeros_likeminimumrnn functional normalize)r4r5r,rr-rr<rweightsfftfreqsmel_ffdifframpsilowerupperenorms r#compute_fbank_matrixrUsF }b A llFCEQJ$78FG"E?H  %u#U E !"Icr "E OOA !3!3A!6 6E6]q E!H$a!e uQU|+^^   e $fnnUE&B   xuQ!,uWf~=>5??1%% N D3, '))&&00Dr0J Nr%spect ref_valueamintop_dbc|dkr td|dkr tdtj|}dtjtj||z|z}|dt jt ||zz}|:|dkr tdtj|||j |z z}|S)ahConvert a power spectrogram (amplitude squared) to decibel (dB) units. The function computes the scaling `10 * log10(x / ref)` in a numerically stable way. Args: spect (Tensor): STFT power spectrogram. ref_value (float, optional): The reference value. If smaller than 1.0, the db level of the signal will be pulled up accordingly. Otherwise, the db level is pushed down. Defaults to 1.0. amin (float, optional): Minimum threshold. Defaults to 1e-10. top_db (Optional[float], optional): Threshold the output at `top_db` below the peak. Defaults to None. Returns: Tensor: Power spectrogram in db scale. Examples: .. code-block:: python >>> import paddle >>> val = 3.0 >>> decibel_paddle = paddle.audio.functional.power_to_db( ... paddle.to_tensor(val)) rzamin must be strictly positivez#ref_value must be strictly positiver(ztop_db must be non-negative) Exceptionr ones_likerrFrmax)rVrWrXrYoneslog_specs r# power_to_dbr`s4 qy899A~=>>   E "Dfll6>>$+u#EFFH tzz#i"6777H  A:9: :>>(DHLLNV4K,LM Or%n_mfccctj||}tj||jd}tjtj t |z |dzz|z}||dz}|jS|dk(sJ|dxxdt jdz zcc<|t jdt |z z}|jS)a{Create a discrete cosine transform(DCT) matrix. Args: n_mfcc (int): Number of mel frequency cepstral coefficients. n_mels (int): Number of mel filterbanks. norm (Optional[str], optional): Normalizaiton type. Defaults to 'ortho'. dtype (str, optional): The data type of the return matrix. Defaults to 'float32'. Returns: Tensor: The DCT matrix with shape `(n_mels, n_mfcc)`. Examples: .. code-block:: python >>> import paddle >>> n_mfcc = 23 >>> n_mels = 257 >>> dct = paddle.audio.functional.create_dct(n_mfcc, n_mels) r/rg?r@orthorr ) rarangerDcosrpir8sqrtT)rar,r<rnkdcts r# create_dctrl/s2  fE*A fE*44Q7A ** %-1s7+a/ C | s  55Lw A# #&& tyyuV},-- 55Lr%)F)@r g@Ffloat32)rn)rmr NFr?rn)r rgT@)rcrn)rtypingrrrrr8boolr$r+r9strr3r;rUr`rlr%r#rss " -25  5%)5 65=5r,1* uf} *$(* 5&=*\ % % % %  %  %  %POOCOOFO4!&F F F F  F E? F  F U F F FV" ) )) ) UO )  )^" $ $ $ 3-$  $  $r%