Spaces:
Running
Running
update
Browse files- examples/sample_filter/bad_case_find.py +8 -8
- examples/sample_filter/correction.py +2 -2
- examples/sample_filter/find_label_error_wav.py +4 -4
- examples/sample_filter/test2.py +8 -8
- examples/sample_filter/wav_find_by_task_excel.py +5 -5
- examples/vm_sound_classification/run.sh +7 -7
- examples/vm_sound_classification/run_batch.sh +24 -24
- examples/vm_sound_classification/step_3_train_model.py +1 -1
- examples/vm_sound_classification/step_6_infer.py +2 -2
- examples/vm_sound_classification/step_7_test_model.py +2 -2
- examples/vm_sound_classification/stop.sh +1 -1
- examples/vm_sound_classification8/run.sh +7 -7
- examples/vm_sound_classification8/stop.sh +1 -1
- install.sh +3 -3
- main.py +2 -2
examples/sample_filter/bad_case_find.py
CHANGED
|
@@ -12,20 +12,20 @@ def get_args():
|
|
| 12 |
parser = argparse.ArgumentParser()
|
| 13 |
parser.add_argument(
|
| 14 |
"--data_dir",
|
| 15 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 16 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 17 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 18 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 19 |
type=str
|
| 20 |
)
|
| 21 |
parser.add_argument(
|
| 22 |
"--keep_dir",
|
| 23 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 24 |
type=str
|
| 25 |
)
|
| 26 |
parser.add_argument(
|
| 27 |
"--trash_dir",
|
| 28 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 29 |
type=str
|
| 30 |
)
|
| 31 |
args = parser.parse_args()
|
|
@@ -50,7 +50,7 @@ def main():
|
|
| 50 |
|
| 51 |
label1, prob1 = client.predict(
|
| 52 |
audio=handle_file(filename),
|
| 53 |
-
# model_name="
|
| 54 |
model_name="voicemail-en-ph-2-ch4",
|
| 55 |
ground_true="Hello!!",
|
| 56 |
api_name="/click_button"
|
|
@@ -59,7 +59,7 @@ def main():
|
|
| 59 |
|
| 60 |
label2, prob2 = client.predict(
|
| 61 |
audio=handle_file(filename),
|
| 62 |
-
# model_name="
|
| 63 |
model_name="sound-8-ch32",
|
| 64 |
ground_true="Hello!!",
|
| 65 |
api_name="/click_button"
|
|
|
|
| 12 |
parser = argparse.ArgumentParser()
|
| 13 |
parser.add_argument(
|
| 14 |
"--data_dir",
|
| 15 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\data",
|
| 16 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\us-3",
|
| 17 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\transfer",
|
| 18 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\id",
|
| 19 |
type=str
|
| 20 |
)
|
| 21 |
parser.add_argument(
|
| 22 |
"--keep_dir",
|
| 23 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\keep",
|
| 24 |
type=str
|
| 25 |
)
|
| 26 |
parser.add_argument(
|
| 27 |
"--trash_dir",
|
| 28 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\trash",
|
| 29 |
type=str
|
| 30 |
)
|
| 31 |
args = parser.parse_args()
|
|
|
|
| 50 |
|
| 51 |
label1, prob1 = client.predict(
|
| 52 |
audio=handle_file(filename),
|
| 53 |
+
# model_name="cc_audio_8-ch32",
|
| 54 |
model_name="voicemail-en-ph-2-ch4",
|
| 55 |
ground_true="Hello!!",
|
| 56 |
api_name="/click_button"
|
|
|
|
| 59 |
|
| 60 |
label2, prob2 = client.predict(
|
| 61 |
audio=handle_file(filename),
|
| 62 |
+
# model_name="cc_audio_8-ch32",
|
| 63 |
model_name="sound-8-ch32",
|
| 64 |
ground_true="Hello!!",
|
| 65 |
api_name="/click_button"
|
examples/sample_filter/correction.py
CHANGED
|
@@ -14,12 +14,12 @@ def get_args():
|
|
| 14 |
parser = argparse.ArgumentParser()
|
| 15 |
parser.add_argument(
|
| 16 |
"--data_dir",
|
| 17 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 18 |
type=str
|
| 19 |
)
|
| 20 |
parser.add_argument(
|
| 21 |
"--correction_dir",
|
| 22 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 23 |
type=str
|
| 24 |
)
|
| 25 |
args = parser.parse_args()
|
|
|
|
| 14 |
parser = argparse.ArgumentParser()
|
| 15 |
parser.add_argument(
|
| 16 |
"--data_dir",
|
| 17 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\wav_finished\en-PH\wav_finished",
|
| 18 |
type=str
|
| 19 |
)
|
| 20 |
parser.add_argument(
|
| 21 |
"--correction_dir",
|
| 22 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\correction",
|
| 23 |
type=str
|
| 24 |
)
|
| 25 |
args = parser.parse_args()
|
examples/sample_filter/find_label_error_wav.py
CHANGED
|
@@ -14,17 +14,17 @@ def get_args():
|
|
| 14 |
parser = argparse.ArgumentParser()
|
| 15 |
parser.add_argument(
|
| 16 |
"--data_dir",
|
| 17 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 18 |
type=str
|
| 19 |
)
|
| 20 |
parser.add_argument(
|
| 21 |
"--keep_dir",
|
| 22 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 23 |
type=str
|
| 24 |
)
|
| 25 |
parser.add_argument(
|
| 26 |
"--trash_dir",
|
| 27 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 28 |
type=str
|
| 29 |
)
|
| 30 |
args = parser.parse_args()
|
|
@@ -50,7 +50,7 @@ def main():
|
|
| 50 |
|
| 51 |
label1, prob1 = client.predict(
|
| 52 |
audio=handle_file(filename),
|
| 53 |
-
# model_name="
|
| 54 |
model_name="voicemail-en-us-2-ch32",
|
| 55 |
ground_true="Hello!!",
|
| 56 |
api_name="/click_button"
|
|
|
|
| 14 |
parser = argparse.ArgumentParser()
|
| 15 |
parser.add_argument(
|
| 16 |
"--data_dir",
|
| 17 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\wav_finished\en-US\wav_finished",
|
| 18 |
type=str
|
| 19 |
)
|
| 20 |
parser.add_argument(
|
| 21 |
"--keep_dir",
|
| 22 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\wav_finished\en-US\keep",
|
| 23 |
type=str
|
| 24 |
)
|
| 25 |
parser.add_argument(
|
| 26 |
"--trash_dir",
|
| 27 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\wav_finished\en-US\trash",
|
| 28 |
type=str
|
| 29 |
)
|
| 30 |
args = parser.parse_args()
|
|
|
|
| 50 |
|
| 51 |
label1, prob1 = client.predict(
|
| 52 |
audio=handle_file(filename),
|
| 53 |
+
# model_name="cc_audio_8-ch32",
|
| 54 |
model_name="voicemail-en-us-2-ch32",
|
| 55 |
ground_true="Hello!!",
|
| 56 |
api_name="/click_button"
|
examples/sample_filter/test2.py
CHANGED
|
@@ -14,21 +14,21 @@ def get_args():
|
|
| 14 |
parser = argparse.ArgumentParser()
|
| 15 |
parser.add_argument(
|
| 16 |
"--data_dir",
|
| 17 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 18 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 19 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 20 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 21 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 22 |
type=str
|
| 23 |
)
|
| 24 |
parser.add_argument(
|
| 25 |
"--keep_dir",
|
| 26 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 27 |
type=str
|
| 28 |
)
|
| 29 |
parser.add_argument(
|
| 30 |
"--trash_dir",
|
| 31 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 32 |
type=str
|
| 33 |
)
|
| 34 |
args = parser.parse_args()
|
|
@@ -53,7 +53,7 @@ def main():
|
|
| 53 |
|
| 54 |
label1, prob1 = client.predict(
|
| 55 |
audio=handle_file(filename),
|
| 56 |
-
# model_name="
|
| 57 |
model_name="voicemail-ms-my-2-ch32",
|
| 58 |
ground_true="Hello!!",
|
| 59 |
api_name="/click_button"
|
|
|
|
| 14 |
parser = argparse.ArgumentParser()
|
| 15 |
parser.add_argument(
|
| 16 |
"--data_dir",
|
| 17 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\data-1",
|
| 18 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\temp-2\temp\VoiceAppVoicemailDetection-1",
|
| 19 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\temp-3\temp\VoiceAppVoicemailDetection-1",
|
| 20 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\temp-4\temp\VoiceAppVoicemailDetection-1",
|
| 21 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\transfer",
|
| 22 |
type=str
|
| 23 |
)
|
| 24 |
parser.add_argument(
|
| 25 |
"--keep_dir",
|
| 26 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\keep-3",
|
| 27 |
type=str
|
| 28 |
)
|
| 29 |
parser.add_argument(
|
| 30 |
"--trash_dir",
|
| 31 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\trash",
|
| 32 |
type=str
|
| 33 |
)
|
| 34 |
args = parser.parse_args()
|
|
|
|
| 53 |
|
| 54 |
label1, prob1 = client.predict(
|
| 55 |
audio=handle_file(filename),
|
| 56 |
+
# model_name="cc_audio_8-ch32",
|
| 57 |
model_name="voicemail-ms-my-2-ch32",
|
| 58 |
ground_true="Hello!!",
|
| 59 |
api_name="/click_button"
|
examples/sample_filter/wav_find_by_task_excel.py
CHANGED
|
@@ -32,15 +32,15 @@ def get_args():
|
|
| 32 |
)
|
| 33 |
parser.add_argument(
|
| 34 |
"--wav_dir",
|
| 35 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 36 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 37 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 38 |
-
# default=r"E:\Users\tianx\HuggingDatasets\
|
| 39 |
type=str
|
| 40 |
)
|
| 41 |
parser.add_argument(
|
| 42 |
"--output_dir",
|
| 43 |
-
default=r"E:\Users\tianx\HuggingDatasets\
|
| 44 |
type=str
|
| 45 |
)
|
| 46 |
args = parser.parse_args()
|
|
|
|
| 32 |
)
|
| 33 |
parser.add_argument(
|
| 34 |
"--wav_dir",
|
| 35 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\phl",
|
| 36 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\temp-2\temp\VoiceAppVoicemailDetection-1",
|
| 37 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\temp-3\temp\VoiceAppVoicemailDetection-1",
|
| 38 |
+
# default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\temp-4\temp\VoiceAppVoicemailDetection-1",
|
| 39 |
type=str
|
| 40 |
)
|
| 41 |
parser.add_argument(
|
| 42 |
"--output_dir",
|
| 43 |
+
default=r"E:\Users\tianx\HuggingDatasets\cc_audio_8\data\transfer",
|
| 44 |
type=str
|
| 45 |
)
|
| 46 |
args = parser.parse_args()
|
examples/vm_sound_classification/run.sh
CHANGED
|
@@ -3,13 +3,13 @@
|
|
| 3 |
: <<'END'
|
| 4 |
|
| 5 |
sh run.sh --stage 0 --stop_stage 1 --system_version windows --file_folder_name file_dir --final_model_name sound-4-ch32 \
|
| 6 |
-
--filename_patterns "E:/Users/tianx/HuggingDatasets/
|
| 7 |
-
E:/Users/tianx/HuggingDatasets/
|
| 8 |
--label_plan 4
|
| 9 |
|
| 10 |
sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name file_dir --final_model_name sound-2-ch32 \
|
| 11 |
-
--filename_patterns "E:/Users/tianx/HuggingDatasets/
|
| 12 |
-
E:/Users/tianx/HuggingDatasets/
|
| 13 |
--label_plan 4
|
| 14 |
|
| 15 |
sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name sound-3-ch32 \
|
|
@@ -91,10 +91,10 @@ $verbose && echo "system_version: ${system_version}"
|
|
| 91 |
$verbose && echo "file_folder_name: ${file_folder_name}"
|
| 92 |
|
| 93 |
if [ $system_version == "windows" ]; then
|
| 94 |
-
alias python3='D:/Users/tianx/PycharmProjects/virtualenv/
|
| 95 |
elif [ $system_version == "centos" ] || [ $system_version == "ubuntu" ]; then
|
| 96 |
-
#source /data/local/bin/
|
| 97 |
-
alias python3='/data/local/bin/
|
| 98 |
fi
|
| 99 |
|
| 100 |
|
|
|
|
| 3 |
: <<'END'
|
| 4 |
|
| 5 |
sh run.sh --stage 0 --stop_stage 1 --system_version windows --file_folder_name file_dir --final_model_name sound-4-ch32 \
|
| 6 |
+
--filename_patterns "E:/Users/tianx/HuggingDatasets/cc_audio_8/data/wav_finished/wav_finished/en-US/wav_finished/*/*.wav \
|
| 7 |
+
E:/Users/tianx/HuggingDatasets/cc_audio_8/data/wav_finished/id-ID/wav_finished/*/*.wav" \
|
| 8 |
--label_plan 4
|
| 9 |
|
| 10 |
sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name file_dir --final_model_name sound-2-ch32 \
|
| 11 |
+
--filename_patterns "E:/Users/tianx/HuggingDatasets/cc_audio_8/data/wav_finished/wav_finished/en-US/wav_finished/*/*.wav \
|
| 12 |
+
E:/Users/tianx/HuggingDatasets/cc_audio_8/data/wav_finished/id-ID/wav_finished/*/*.wav" \
|
| 13 |
--label_plan 4
|
| 14 |
|
| 15 |
sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name sound-3-ch32 \
|
|
|
|
| 91 |
$verbose && echo "file_folder_name: ${file_folder_name}"
|
| 92 |
|
| 93 |
if [ $system_version == "windows" ]; then
|
| 94 |
+
alias python3='D:/Users/tianx/PycharmProjects/virtualenv/cc_audio_8/Scripts/python.exe'
|
| 95 |
elif [ $system_version == "centos" ] || [ $system_version == "ubuntu" ]; then
|
| 96 |
+
#source /data/local/bin/cc_audio_8/bin/activate
|
| 97 |
+
alias python3='/data/local/bin/cc_audio_8/bin/python3'
|
| 98 |
fi
|
| 99 |
|
| 100 |
|
examples/vm_sound_classification/run_batch.sh
CHANGED
|
@@ -124,73 +124,73 @@ sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name fi
|
|
| 124 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-PH/wav_finished/*/*.wav" \
|
| 125 |
--label_plan 2-voicemail \
|
| 126 |
--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 127 |
-
--pretrained_model "/data/tianxing/PycharmProjects/
|
| 128 |
|
| 129 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-sg-2-ch4 \
|
| 130 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-SG/wav_finished/*/*.wav" \
|
| 131 |
#--label_plan 2-voicemail \
|
| 132 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 133 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 134 |
#
|
| 135 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-us-2-ch4 \
|
| 136 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-US/wav_finished/*/*.wav" \
|
| 137 |
#--label_plan 2-voicemail \
|
| 138 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 139 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 140 |
#
|
| 141 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-mx-2-ch4 \
|
| 142 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-MX/wav_finished/*/*.wav" \
|
| 143 |
#--label_plan 2-voicemail \
|
| 144 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 145 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 146 |
#
|
| 147 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-pe-2-ch4 \
|
| 148 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-PE/wav_finished/*/*.wav" \
|
| 149 |
#--label_plan 2-voicemail \
|
| 150 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 151 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 152 |
#
|
| 153 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-id-id-2-ch4 \
|
| 154 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/id-ID/wav_finished/*/*.wav" \
|
| 155 |
#--label_plan 2-voicemail \
|
| 156 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 157 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 158 |
#
|
| 159 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ja-jp-2-ch4 \
|
| 160 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ja-JP/wav_finished/*/*.wav" \
|
| 161 |
#--label_plan 2-voicemail \
|
| 162 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 163 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 164 |
#
|
| 165 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ko-kr-2-ch4 \
|
| 166 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ko-KR/wav_finished/*/*.wav" \
|
| 167 |
#--label_plan 2-voicemail \
|
| 168 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 169 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 170 |
#
|
| 171 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ms-my-2-ch4 \
|
| 172 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ms-MY/wav_finished/*/*.wav" \
|
| 173 |
#--label_plan 2-voicemail \
|
| 174 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 175 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 176 |
#
|
| 177 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-pt-br-2-ch4 \
|
| 178 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/pt-BR/wav_finished/*/*.wav" \
|
| 179 |
#--label_plan 2-voicemail \
|
| 180 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 181 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 182 |
#
|
| 183 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-th-th-2-ch4 \
|
| 184 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/th-TH/wav_finished/*/*.wav" \
|
| 185 |
#--label_plan 2-voicemail \
|
| 186 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 187 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 188 |
#
|
| 189 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-zh-tw-2-ch4 \
|
| 190 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/zh-TW/wav_finished/*/*.wav" \
|
| 191 |
#--label_plan 2-voicemail \
|
| 192 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 193 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 194 |
|
| 195 |
|
| 196 |
# voicemail ch32
|
|
@@ -199,70 +199,70 @@ sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name fi
|
|
| 199 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-PH/wav_finished/*/*.wav" \
|
| 200 |
--label_plan 2-voicemail \
|
| 201 |
--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 202 |
-
--pretrained_model "/data/tianxing/PycharmProjects/
|
| 203 |
|
| 204 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-sg-2-ch32 \
|
| 205 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-SG/wav_finished/*/*.wav" \
|
| 206 |
#--label_plan 2-voicemail \
|
| 207 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 208 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 209 |
#
|
| 210 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-us-2-ch32 \
|
| 211 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-US/wav_finished/*/*.wav" \
|
| 212 |
#--label_plan 2-voicemail \
|
| 213 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 214 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 215 |
#
|
| 216 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-mx-2-ch32 \
|
| 217 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-MX/wav_finished/*/*.wav" \
|
| 218 |
#--label_plan 2-voicemail \
|
| 219 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 220 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 221 |
#
|
| 222 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-pe-2-ch32 \
|
| 223 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-PE/wav_finished/*/*.wav" \
|
| 224 |
#--label_plan 2-voicemail \
|
| 225 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 226 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 227 |
#
|
| 228 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-id-id-2-ch32 \
|
| 229 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/id-ID/wav_finished/*/*.wav" \
|
| 230 |
#--label_plan 2-voicemail \
|
| 231 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 232 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 233 |
#
|
| 234 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ja-jp-2-ch32 \
|
| 235 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ja-JP/wav_finished/*/*.wav" \
|
| 236 |
#--label_plan 2-voicemail \
|
| 237 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 238 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 239 |
#
|
| 240 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ko-kr-2-ch32 \
|
| 241 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ko-KR/wav_finished/*/*.wav" \
|
| 242 |
#--label_plan 2-voicemail \
|
| 243 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 244 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 245 |
#
|
| 246 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ms-my-2-ch32 \
|
| 247 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ms-MY/wav_finished/*/*.wav" \
|
| 248 |
#--label_plan 2-voicemail \
|
| 249 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 250 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 251 |
#
|
| 252 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-pt-br-2-ch32 \
|
| 253 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/pt-BR/wav_finished/*/*.wav" \
|
| 254 |
#--label_plan 2-voicemail \
|
| 255 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 256 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 257 |
#
|
| 258 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-th-th-2-ch32 \
|
| 259 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/th-TH/wav_finished/*/*.wav" \
|
| 260 |
#--label_plan 2-voicemail \
|
| 261 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 262 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
| 263 |
#
|
| 264 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-zh-tw-2-ch32 \
|
| 265 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/zh-TW/wav_finished/*/*.wav" \
|
| 266 |
#--label_plan 2-voicemail \
|
| 267 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 268 |
-
#--pretrained_model "/data/tianxing/PycharmProjects/
|
|
|
|
| 124 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-PH/wav_finished/*/*.wav" \
|
| 125 |
--label_plan 2-voicemail \
|
| 126 |
--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 127 |
+
--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 128 |
|
| 129 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-sg-2-ch4 \
|
| 130 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-SG/wav_finished/*/*.wav" \
|
| 131 |
#--label_plan 2-voicemail \
|
| 132 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 133 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 134 |
#
|
| 135 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-us-2-ch4 \
|
| 136 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-US/wav_finished/*/*.wav" \
|
| 137 |
#--label_plan 2-voicemail \
|
| 138 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 139 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 140 |
#
|
| 141 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-mx-2-ch4 \
|
| 142 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-MX/wav_finished/*/*.wav" \
|
| 143 |
#--label_plan 2-voicemail \
|
| 144 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 145 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 146 |
#
|
| 147 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-pe-2-ch4 \
|
| 148 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-PE/wav_finished/*/*.wav" \
|
| 149 |
#--label_plan 2-voicemail \
|
| 150 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 151 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 152 |
#
|
| 153 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-id-id-2-ch4 \
|
| 154 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/id-ID/wav_finished/*/*.wav" \
|
| 155 |
#--label_plan 2-voicemail \
|
| 156 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 157 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 158 |
#
|
| 159 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ja-jp-2-ch4 \
|
| 160 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ja-JP/wav_finished/*/*.wav" \
|
| 161 |
#--label_plan 2-voicemail \
|
| 162 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 163 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 164 |
#
|
| 165 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ko-kr-2-ch4 \
|
| 166 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ko-KR/wav_finished/*/*.wav" \
|
| 167 |
#--label_plan 2-voicemail \
|
| 168 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 169 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 170 |
#
|
| 171 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ms-my-2-ch4 \
|
| 172 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ms-MY/wav_finished/*/*.wav" \
|
| 173 |
#--label_plan 2-voicemail \
|
| 174 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 175 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 176 |
#
|
| 177 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-pt-br-2-ch4 \
|
| 178 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/pt-BR/wav_finished/*/*.wav" \
|
| 179 |
#--label_plan 2-voicemail \
|
| 180 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 181 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 182 |
#
|
| 183 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-th-th-2-ch4 \
|
| 184 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/th-TH/wav_finished/*/*.wav" \
|
| 185 |
#--label_plan 2-voicemail \
|
| 186 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 187 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 188 |
#
|
| 189 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-zh-tw-2-ch4 \
|
| 190 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/zh-TW/wav_finished/*/*.wav" \
|
| 191 |
#--label_plan 2-voicemail \
|
| 192 |
#--config_file "yaml/conv2d-classifier-2-ch4.yaml" \
|
| 193 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch4.zip"
|
| 194 |
|
| 195 |
|
| 196 |
# voicemail ch32
|
|
|
|
| 199 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-PH/wav_finished/*/*.wav" \
|
| 200 |
--label_plan 2-voicemail \
|
| 201 |
--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 202 |
+
--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 203 |
|
| 204 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-sg-2-ch32 \
|
| 205 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-SG/wav_finished/*/*.wav" \
|
| 206 |
#--label_plan 2-voicemail \
|
| 207 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 208 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 209 |
#
|
| 210 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-en-us-2-ch32 \
|
| 211 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/en-US/wav_finished/*/*.wav" \
|
| 212 |
#--label_plan 2-voicemail \
|
| 213 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 214 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 215 |
#
|
| 216 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-mx-2-ch32 \
|
| 217 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-MX/wav_finished/*/*.wav" \
|
| 218 |
#--label_plan 2-voicemail \
|
| 219 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 220 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 221 |
#
|
| 222 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-es-pe-2-ch32 \
|
| 223 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/es-PE/wav_finished/*/*.wav" \
|
| 224 |
#--label_plan 2-voicemail \
|
| 225 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 226 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 227 |
#
|
| 228 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-id-id-2-ch32 \
|
| 229 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/id-ID/wav_finished/*/*.wav" \
|
| 230 |
#--label_plan 2-voicemail \
|
| 231 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 232 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 233 |
#
|
| 234 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ja-jp-2-ch32 \
|
| 235 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ja-JP/wav_finished/*/*.wav" \
|
| 236 |
#--label_plan 2-voicemail \
|
| 237 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 238 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 239 |
#
|
| 240 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ko-kr-2-ch32 \
|
| 241 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ko-KR/wav_finished/*/*.wav" \
|
| 242 |
#--label_plan 2-voicemail \
|
| 243 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 244 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 245 |
#
|
| 246 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-ms-my-2-ch32 \
|
| 247 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/ms-MY/wav_finished/*/*.wav" \
|
| 248 |
#--label_plan 2-voicemail \
|
| 249 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 250 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 251 |
#
|
| 252 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-pt-br-2-ch32 \
|
| 253 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/pt-BR/wav_finished/*/*.wav" \
|
| 254 |
#--label_plan 2-voicemail \
|
| 255 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 256 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 257 |
#
|
| 258 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-th-th-2-ch32 \
|
| 259 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/th-TH/wav_finished/*/*.wav" \
|
| 260 |
#--label_plan 2-voicemail \
|
| 261 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 262 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
| 263 |
#
|
| 264 |
#sh run.sh --stage 0 --stop_stage 6 --system_version centos --file_folder_name file_dir --final_model_name voicemail-zh-tw-2-ch32 \
|
| 265 |
#--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/zh-TW/wav_finished/*/*.wav" \
|
| 266 |
#--label_plan 2-voicemail \
|
| 267 |
#--config_file "yaml/conv2d-classifier-2-ch32.yaml" \
|
| 268 |
+
#--pretrained_model "/data/tianxing/PycharmProjects/cc_audio_8/trained_models/voicemail-2-ch32.zip"
|
examples/vm_sound_classification/step_3_train_model.py
CHANGED
|
@@ -188,7 +188,7 @@ def main():
|
|
| 188 |
logger.info(f"load pretrained model state dict from: {args.pretrained_model}")
|
| 189 |
pretrained_model = Path(args.pretrained_model)
|
| 190 |
with zipfile.ZipFile(pretrained_model.as_posix(), "r") as f_zip:
|
| 191 |
-
out_root = Path(tempfile.gettempdir()) / "
|
| 192 |
# print(out_root.as_posix())
|
| 193 |
if out_root.exists():
|
| 194 |
shutil.rmtree(out_root.as_posix())
|
|
|
|
| 188 |
logger.info(f"load pretrained model state dict from: {args.pretrained_model}")
|
| 189 |
pretrained_model = Path(args.pretrained_model)
|
| 190 |
with zipfile.ZipFile(pretrained_model.as_posix(), "r") as f_zip:
|
| 191 |
+
out_root = Path(tempfile.gettempdir()) / "cc_audio_8"
|
| 192 |
# print(out_root.as_posix())
|
| 193 |
if out_root.exists():
|
| 194 |
shutil.rmtree(out_root.as_posix())
|
examples/vm_sound_classification/step_6_infer.py
CHANGED
|
@@ -22,7 +22,7 @@ def get_args():
|
|
| 22 |
parser = argparse.ArgumentParser()
|
| 23 |
parser.add_argument(
|
| 24 |
"--model_file",
|
| 25 |
-
default=(project_path / "trained_models/
|
| 26 |
type=str
|
| 27 |
)
|
| 28 |
parser.add_argument(
|
|
@@ -45,7 +45,7 @@ def main():
|
|
| 45 |
device = torch.device(args.device)
|
| 46 |
|
| 47 |
with zipfile.ZipFile(model_file, "r") as f_zip:
|
| 48 |
-
out_root = Path(tempfile.gettempdir()) / "
|
| 49 |
print(out_root.as_posix())
|
| 50 |
if out_root.exists():
|
| 51 |
shutil.rmtree(out_root.as_posix())
|
|
|
|
| 22 |
parser = argparse.ArgumentParser()
|
| 23 |
parser.add_argument(
|
| 24 |
"--model_file",
|
| 25 |
+
default=(project_path / "trained_models/cc_audio_8.zip").as_posix(),
|
| 26 |
type=str
|
| 27 |
)
|
| 28 |
parser.add_argument(
|
|
|
|
| 45 |
device = torch.device(args.device)
|
| 46 |
|
| 47 |
with zipfile.ZipFile(model_file, "r") as f_zip:
|
| 48 |
+
out_root = Path(tempfile.gettempdir()) / "cc_audio_8"
|
| 49 |
print(out_root.as_posix())
|
| 50 |
if out_root.exists():
|
| 51 |
shutil.rmtree(out_root.as_posix())
|
examples/vm_sound_classification/step_7_test_model.py
CHANGED
|
@@ -23,7 +23,7 @@ def get_args():
|
|
| 23 |
parser = argparse.ArgumentParser()
|
| 24 |
parser.add_argument(
|
| 25 |
"--model_file",
|
| 26 |
-
default=(project_path / "trained_models/
|
| 27 |
type=str
|
| 28 |
)
|
| 29 |
parser.add_argument(
|
|
@@ -46,7 +46,7 @@ def main():
|
|
| 46 |
device = torch.device(args.device)
|
| 47 |
|
| 48 |
with zipfile.ZipFile(model_file, "r") as f_zip:
|
| 49 |
-
out_root = Path(tempfile.gettempdir()) / "
|
| 50 |
print(out_root)
|
| 51 |
if out_root.exists():
|
| 52 |
shutil.rmtree(out_root.as_posix())
|
|
|
|
| 23 |
parser = argparse.ArgumentParser()
|
| 24 |
parser.add_argument(
|
| 25 |
"--model_file",
|
| 26 |
+
default=(project_path / "trained_models/cc_audio_8.zip").as_posix(),
|
| 27 |
type=str
|
| 28 |
)
|
| 29 |
parser.add_argument(
|
|
|
|
| 46 |
device = torch.device(args.device)
|
| 47 |
|
| 48 |
with zipfile.ZipFile(model_file, "r") as f_zip:
|
| 49 |
+
out_root = Path(tempfile.gettempdir()) / "cc_audio_8"
|
| 50 |
print(out_root)
|
| 51 |
if out_root.exists():
|
| 52 |
shutil.rmtree(out_root.as_posix())
|
examples/vm_sound_classification/stop.sh
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
|
| 3 |
-
kill -9 `ps -aef | grep '
|
|
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
|
| 3 |
+
kill -9 `ps -aef | grep 'cc_audio_8/bin/python3' | grep -v grep | awk '{print $2}' | sed 's/\n/ /'`
|
examples/vm_sound_classification8/run.sh
CHANGED
|
@@ -2,18 +2,18 @@
|
|
| 2 |
|
| 3 |
: <<'END'
|
| 4 |
|
| 5 |
-
sh run.sh --stage 3 --stop_stage 3 --system_version windows --file_folder_name file_dir --final_model_name
|
| 6 |
--filename_patterns "E:/programmer/asr_datasets/voicemail/wav_finished/en-US/wav_finished/*/*.wav \
|
| 7 |
E:/programmer/asr_datasets/voicemail/wav_finished/id-ID/wav_finished/*/*.wav" \
|
| 8 |
|
| 9 |
|
| 10 |
-
sh run.sh --stage 3 --stop_stage 3 --system_version windows --file_folder_name file_dir --final_model_name
|
| 11 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav"
|
| 12 |
|
| 13 |
-
sh run.sh --stage 4 --stop_stage 4 --system_version windows --file_folder_name file_dir --final_model_name
|
| 14 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav"
|
| 15 |
|
| 16 |
-
sh run.sh --stage 4 --stop_stage 4 --system_version centos --file_folder_name file_dir --final_model_name
|
| 17 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav"
|
| 18 |
|
| 19 |
|
|
@@ -90,10 +90,10 @@ $verbose && echo "system_version: ${system_version}"
|
|
| 90 |
$verbose && echo "file_folder_name: ${file_folder_name}"
|
| 91 |
|
| 92 |
if [ $system_version == "windows" ]; then
|
| 93 |
-
alias python3='D:/Users/tianx/PycharmProjects/virtualenv/
|
| 94 |
elif [ $system_version == "centos" ] || [ $system_version == "ubuntu" ]; then
|
| 95 |
-
#source /data/local/bin/
|
| 96 |
-
alias python3='/data/local/bin/
|
| 97 |
fi
|
| 98 |
|
| 99 |
|
|
|
|
| 2 |
|
| 3 |
: <<'END'
|
| 4 |
|
| 5 |
+
sh run.sh --stage 3 --stop_stage 3 --system_version windows --file_folder_name file_dir --final_model_name cc_audio_8 \
|
| 6 |
--filename_patterns "E:/programmer/asr_datasets/voicemail/wav_finished/en-US/wav_finished/*/*.wav \
|
| 7 |
E:/programmer/asr_datasets/voicemail/wav_finished/id-ID/wav_finished/*/*.wav" \
|
| 8 |
|
| 9 |
|
| 10 |
+
sh run.sh --stage 3 --stop_stage 3 --system_version windows --file_folder_name file_dir --final_model_name cc_audio_8 \
|
| 11 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav"
|
| 12 |
|
| 13 |
+
sh run.sh --stage 4 --stop_stage 4 --system_version windows --file_folder_name file_dir --final_model_name cc_audio_8 \
|
| 14 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav"
|
| 15 |
|
| 16 |
+
sh run.sh --stage 4 --stop_stage 4 --system_version centos --file_folder_name file_dir --final_model_name cc_audio_8 \
|
| 17 |
--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav"
|
| 18 |
|
| 19 |
|
|
|
|
| 90 |
$verbose && echo "file_folder_name: ${file_folder_name}"
|
| 91 |
|
| 92 |
if [ $system_version == "windows" ]; then
|
| 93 |
+
alias python3='D:/Users/tianx/PycharmProjects/virtualenv/cc_audio_8/Scripts/python.exe'
|
| 94 |
elif [ $system_version == "centos" ] || [ $system_version == "ubuntu" ]; then
|
| 95 |
+
#source /data/local/bin/cc_audio_8/bin/activate
|
| 96 |
+
alias python3='/data/local/bin/cc_audio_8/bin/python3'
|
| 97 |
fi
|
| 98 |
|
| 99 |
|
examples/vm_sound_classification8/stop.sh
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
|
| 3 |
-
kill -9 `ps -aef | grep '
|
|
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
|
| 3 |
+
kill -9 `ps -aef | grep 'cc_audio_8/bin/python3' | grep -v grep | awk '{print $2}' | sed 's/\n/ /'`
|
install.sh
CHANGED
|
@@ -54,11 +54,11 @@ fi
|
|
| 54 |
if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
|
| 55 |
$verbose && echo "stage 2: create virtualenv"
|
| 56 |
|
| 57 |
-
# /usr/local/python-3.6.5/bin/virtualenv
|
| 58 |
-
# source /data/local/bin/
|
| 59 |
/usr/local/python-${python_version}/bin/pip3 install virtualenv
|
| 60 |
mkdir -p /data/local/bin
|
| 61 |
cd /data/local/bin || exit 1;
|
| 62 |
-
/usr/local/python-${python_version}/bin/virtualenv
|
| 63 |
|
| 64 |
fi
|
|
|
|
| 54 |
if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
|
| 55 |
$verbose && echo "stage 2: create virtualenv"
|
| 56 |
|
| 57 |
+
# /usr/local/python-3.6.5/bin/virtualenv cc_audio_8
|
| 58 |
+
# source /data/local/bin/cc_audio_8/bin/activate
|
| 59 |
/usr/local/python-${python_version}/bin/pip3 install virtualenv
|
| 60 |
mkdir -p /data/local/bin
|
| 61 |
cd /data/local/bin || exit 1;
|
| 62 |
+
/usr/local/python-${python_version}/bin/virtualenv cc_audio_8
|
| 63 |
|
| 64 |
fi
|
main.py
CHANGED
|
@@ -28,7 +28,7 @@ def get_args():
|
|
| 28 |
)
|
| 29 |
parser.add_argument(
|
| 30 |
"--models_repo_id",
|
| 31 |
-
default="qgyd2021/
|
| 32 |
type=str
|
| 33 |
)
|
| 34 |
parser.add_argument(
|
|
@@ -54,7 +54,7 @@ def get_args():
|
|
| 54 |
@lru_cache(maxsize=100)
|
| 55 |
def load_model(model_file: Path):
|
| 56 |
with zipfile.ZipFile(model_file, "r") as f_zip:
|
| 57 |
-
out_root = Path(tempfile.gettempdir()) / "
|
| 58 |
if out_root.exists():
|
| 59 |
shutil.rmtree(out_root.as_posix())
|
| 60 |
out_root.mkdir(parents=True, exist_ok=True)
|
|
|
|
| 28 |
)
|
| 29 |
parser.add_argument(
|
| 30 |
"--models_repo_id",
|
| 31 |
+
default="qgyd2021/cc_audio_8",
|
| 32 |
type=str
|
| 33 |
)
|
| 34 |
parser.add_argument(
|
|
|
|
| 54 |
@lru_cache(maxsize=100)
|
| 55 |
def load_model(model_file: Path):
|
| 56 |
with zipfile.ZipFile(model_file, "r") as f_zip:
|
| 57 |
+
out_root = Path(tempfile.gettempdir()) / "cc_audio_8"
|
| 58 |
if out_root.exists():
|
| 59 |
shutil.rmtree(out_root.as_posix())
|
| 60 |
out_root.mkdir(parents=True, exist_ok=True)
|