update vox plda recipe

wenet-e2e · Jul 19, 2023 · ebfa7d0 · ebfa7d0
1 parent 602cfee
commit ebfa7d0
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 3 deletions.
diff --git a/examples/__init__.py b/examples/__init__.py
diff --git a/examples/sre/__init__.py b/examples/sre/__init__.py
diff --git a/examples/sre/v2/__init__.py b/examples/sre/v2/__init__.py
diff --git a/examples/voxceleb/v2/local/score_plda.sh b/examples/voxceleb/v2/local/score_plda.sh
@@ -26,14 +26,12 @@ stop_stage=-1
 
 if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
   echo "train the plda model ..."
-  mkdir -p ${exp_dir}/scores
   python wespeaker/bin/train_plda.py \
     --exp_dir ${exp_dir} \
     --scp_path ${exp_dir}/embeddings/vox2_dev/xvector.scp \
     --utt2spk ${data}/vox2_dev/utt2spk \
     --indim 256 \
-    --iter 5 \
-    --type '2cov'
+    --iter 5
   echo "plda training finished"
 fi
 
@@ -52,6 +50,22 @@ if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
   done
 fi
 
+if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
+  echo "apply plda scoring ..."
+  mkdir -p ${exp_dir}/scores
+  trials_dir=${data}/vox1/trials
+  for x in $trials; do
+    echo "scoring on " $x
+    python wespeaker/bin/eval_plda.py \
+      --enroll_scp_path ${exp_dir}/embeddings/vox1/xvector.scp \
+      --test_scp_path ${exp_dir}/embeddings/vox1/xvector.scp \
+      --utt2spk <(cat ${data}/vox1/utt2spk | awk '{print $1, $1}') \
+      --trial ${trials_dir}/${x} \
+      --score_path ${exp_dir}/scores/${x}.pldascore \
+      --model_path ${exp_dir}/plda
+  done
+fi
+
 if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
   echo "compute metrics (EER/minDCF) ..."
   scores_dir=${exp_dir}/scores