| import os | |
| from pathlib import Path | |
| import numpy as np | |
| from tqdm import tqdm | |
| from infer_tools import infer_tool | |
| from preprocessing.hubertinfer import HubertEncoder | |
| # hubert_mode可选——"soft_hubert"、"cn_hubert" | |
| hubert_model = HubertEncoder(hubert_mode='cn_hubert') | |
| # 自动搜索batch文件夹下所有wav文件，可自行更改路径 | |
| wav_paths = infer_tool.get_end_file("./batch", "wav") | |
| with tqdm(total=len(wav_paths)) as p_bar: | |
| p_bar.set_description('Processing') | |
| for wav_path in wav_paths: | |
| npy_path = Path(wav_path).with_suffix(".npy") | |
| if not os.path.exists(npy_path): | |
| np.save(str(npy_path), hubert_model.encode(wav_path)) | |
| p_bar.update(1) | |