aiqcamp commited on
Commit
9ca4ddb
·
verified ·
1 Parent(s): 411c5f2

Delete tools/cmd-denoise.py

Browse files
Files changed (1) hide show
  1. tools/cmd-denoise.py +0 -33
tools/cmd-denoise.py DELETED
@@ -1,33 +0,0 @@
1
- import os,argparse
2
- import traceback
3
-
4
- from modelscope.pipelines import pipeline
5
- from modelscope.utils.constant import Tasks
6
- from tqdm import tqdm
7
-
8
- path_denoise = 'tools/denoise-model/speech_frcrn_ans_cirm_16k'
9
- path_denoise = path_denoise if os.path.exists(path_denoise) else "damo/speech_frcrn_ans_cirm_16k"
10
- ans = pipeline(Tasks.acoustic_noise_suppression,model=path_denoise)
11
- def execute_denoise(input_folder,output_folder):
12
- os.makedirs(output_folder,exist_ok=True)
13
- # print(input_folder)
14
- # print(list(os.listdir(input_folder).sort()))
15
- for name in tqdm(os.listdir(input_folder)):
16
- try:
17
- ans("%s/%s"%(input_folder,name),output_path='%s/%s'%(output_folder,name))
18
- except:
19
- traceback.print_exc()
20
-
21
- if __name__ == '__main__':
22
- parser = argparse.ArgumentParser()
23
- parser.add_argument("-i", "--input_folder", type=str, required=True,
24
- help="Path to the folder containing WAV files.")
25
- parser.add_argument("-o", "--output_folder", type=str, required=True,
26
- help="Output folder to store transcriptions.")
27
- parser.add_argument("-p", "--precision", type=str, default='float16', choices=['float16','float32'],
28
- help="fp16 or fp32")#还没接入
29
- cmd = parser.parse_args()
30
- execute_denoise(
31
- input_folder = cmd.input_folder,
32
- output_folder = cmd.output_folder,
33
- )