| | |
|---|
| | | if key == "config_path": |
|---|
| | | with open(str(value), 'rb') as f: |
|---|
| | | config = yaml.load(f, Loader=yaml.Loader) |
|---|
| | | if key == "cmvn_path": |
|---|
| | | cmvn_path = str(value) |
|---|
| | | |
|---|
| | | opts = kaldifeat.FbankOptions() |
|---|
| | | opts.frame_opts.dither = 1.0 # TODO: 0.0 or 1.0 |
|---|
| | | opts.frame_opts.window_type = config['WavFrontend']['frontend_conf']['window'] |
|---|
| | | opts.mel_opts.num_bins = int(config['WavFrontend']['frontend_conf']['n_mels']) |
|---|
| | | opts.frame_opts.frame_shift_ms = float(config['WavFrontend']['frontend_conf']['frame_shift']) |
|---|
| | | opts.frame_opts.frame_length_ms = float(config['WavFrontend']['frontend_conf']['frame_length']) |
|---|
| | | opts.frame_opts.samp_freq = int(config['WavFrontend']['frontend_conf']['fs']) |
|---|
| | | opts.frame_opts.window_type = config['frontend_conf']['window'] |
|---|
| | | opts.mel_opts.num_bins = int(config['frontend_conf']['n_mels']) |
|---|
| | | opts.frame_opts.frame_shift_ms = float(config['frontend_conf']['frame_shift']) |
|---|
| | | opts.frame_opts.frame_length_ms = float(config['frontend_conf']['frame_length']) |
|---|
| | | opts.frame_opts.samp_freq = int(config['frontend_conf']['fs']) |
|---|
| | | opts.device = torch.device(self.device) |
|---|
| | | self.opts = opts |
|---|
| | | self.feature_extractor = Fbank(self.opts) |
|---|
| | | self.feature_size = opts.mel_opts.num_bins |
|---|
| | | |
|---|
| | | self.frontend = WavFrontend( |
|---|
| | | cmvn_file=config['WavFrontend']['cmvn_file'], |
|---|
| | | **config['WavFrontend']['frontend_conf']) |
|---|
| | | cmvn_file=cmvn_path, |
|---|
| | | **config['frontend_conf']) |
|---|
| | | |
|---|
| | | def extract_feat(self, |
|---|
| | | waveform_list: List[np.ndarray] |
|---|
