Delete tools/cmd-denoise.py
Browse files- 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 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|