o
    
jg                     @   sn   d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm	Z	 d dl
mZ dd Zedkr5e  dS dS )	    N)tqdm)load_config)load_tts_samples)AudioProcessorc                  C   sv  t jdd} | jdtdd | jdtdd | jdtd	d
d |  \}}t|j}|j|dd d	|j_d|j_	t
d$i |j }|jrUtjtj|jdddd}nt|jd }tdt| d d}d}d}d}	d}
t|D ]C}|t|tr|n|d }||}||}|
|jd 7 }
||d7 }||d7 }||d jdd7 }|	|d jdd7 }	qt||
 }t||
 |d  }||
 }t|	|
 |d  }|j}i }||d< ||d< ||d< ||d< td|   td|   td|   td |   ||j_	d|j_|j` |j`!|j`"|j`#|j |d!< tj$||dd" td#|  dS )%zRun preprocessing process.z3Compute mean and variance of spectrogtram features.)descriptionconfig_pathz:TTS config file path to define audio processin parameters.)typehelpout_pathz#save path (directory and filename).z--data_pathFzBfolder including the target set of wavs overriding dataset config.)r   requiredr	   T)relaxed_parserNz**z*.wav)	recursiver   z > There are z files.
audio_file      )axismel_meanmel_stdlinear_mean
linear_stdz > Avg mel spec mean: z > Avg mel spec scale: z > Avg linear spec mean: z > Avg linear spec scale: audio_config)allow_picklez > stats saved to  )%argparseArgumentParseradd_argumentstrparse_known_argsr   r   audiosignal_norm
stats_pathr   to_dict	data_pathglobospathjoinr   datasetsprintlenr   load_wav
isinstancespectrogrammelspectrogramshapesumnpsqrtr
   meanmax_normmin_level_dbsymmetric_norm	clip_normsave)parserargs	overridesCONFIGapdataset_itemsmel_summel_square_sum
linear_sumlinear_square_sumNitemwavlinearmelr   	mel_scaler   linear_scaleoutput_file_pathstatsr   r   M/home/kuhnn/.local/lib/python3.10/site-packages/TTS/bin/compute_statistics.pymain   sp   


rL   __main__)r   r#   r$   numpyr0   r   
TTS.configr   TTS.tts.datasetsr   TTS.utils.audior   rL   __name__r   r   r   rK   <module>   s   N
