| | |
| | | echo "Downloading Pre-trained model..." |
| | | git clone https://www.modelscope.cn/damo/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch.git |
| | | git clone https://www.modelscope.cn/damo/speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch.git |
| | | ln -s speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/sv.pth ./sv.pth |
| | | ln -s speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/sv.pb ./sv.pb |
| | | cp speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/sv.yaml ./sv.yaml |
| | | ln -s speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch/sond.pth ./sond.pth |
| | | ln -s speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch/sond.pb ./sond.pb |
| | | cp speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch/sond_fbank.yaml ./sond_fbank.yaml |
| | | cp speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch/sond.yaml ./sond.yaml |
| | | echo "Done." |
| | |
| | | |
| | | if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then |
| | | echo "Calculating diarization results..." |
| | | python infer_alimeeting_test.py sond_fbank.yaml sond.pth outputs |
| | | python infer_alimeeting_test.py sond_fbank.yaml sond.pb outputs |
| | | python local/convert_label_to_rttm.py \ |
| | | outputs/labels.txt \ |
| | | data/test_rmsil/raw_rmsil_map.scp \ |