handle nan's in vmafs

master-of-zen · May 18, 2020 · 6984fbe · 6984fbe
1 parent a484130
commit 6984fbe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/av1an.py b/av1an.py
@@ -83,7 +83,7 @@ def read_vmaf_xml(file):
         # Data
         x = [x for x in range(len(vmafs))]
         mean = round(sum(vmafs) / len(vmafs), 3)
-        calc = [x for x in vmafs if isinstance(x, float)]
+        calc = [x for x in vmafs if isinstance(x, float) and not isnan(x)]
         perc_1 = round(np.percentile(calc, 1), 3)
         perc_25 = round(np.percentile(calc, 25), 3)
         perc_75 = round(np.percentile(calc, 75), 3)