shithub: aubio

Info  •  Files  •  Log  •  Branches

src/tempo/beattracking.c: improve confidence values by using interpolated peak magnitude and avoiding nan

--- a/src/tempo/beattracking.c

+++ b/src/tempo/beattracking.c

@@ -421,8 +421,10 @@

 aubio_beattracking_get_confidence (aubio_beattracking_t * bt)

 {

   if (bt->gp) {

-    return fvec_max (bt->acfout) / fvec_sum(bt->acfout);

-  } else {

-    return 0.;

+    smpl_t acf_sum = fvec_sum(bt->acfout);

+    if (acf_sum != 0.) {

+      return fvec_quadratic_peak_mag (bt->acfout, bt->gp) / acf_sum;

+    }

   }

+  return 0.;

 }