face file

1 files changed, 27 insertions, 21 deletions

diff --git a/data/face-frame-recognition-accuracy.py b/data/face-frame-recognition-accuracy.py
index 0d34cf9..3ff0aa8 100755
--- a/data/face-frame-recognition-accuracy.py
+++ b/data/face-frame-recognition-accuracy.py
@@ -1,4 +1,5 @@
 #!/usr/bin/python
+import numpy as np
 import os
 import sys
 import pickle
@@ -22,8 +23,8 @@ for line in open(sys.argv[1]):
     except:
         continue
     user = line[1]
-    if run not in runs:
-        runs[run] = 0
+    if run not in labels:
+        #runs[run] = 0
         labels[run] = users.index(user) + 1
 
 for line in open(sys.argv[2]):
@@ -41,23 +42,28 @@ for line in open(sys.argv[2]):
     conf[run] = float(line[7])
     prun = run
 
-t=0.0
-tp=0.0
-fp=0.0
-fn=0.0
-for (k,v) in runs.items():
-    #print v,labels[k]
-    if v == 0 or conf[k] < thresh:
-        fn += 1
-    elif v != labels[k]:
-        fp += 1
-    else:
-        tp += 1
-    t += 1
-#print runs[167],labels[167]
-#print tp,fp,fn
-#print("Precision: ",tp/(tp+fp))
-#print("False positives: ",fp/(tp+fp))
-#print("Recall: ",1.0-fn/t)
-print str(tp/(tp+fp))+","+str(1.0-fn/t)
+for i in range(999)+list(np.arange(999,1000,0.01)):
+    thresh = i/10
+    t=0.0
+    tp=0.0
+    fp=0.0
+    fn=0.0
+    for (k,v) in runs.items():
+        #print v,labels[k]
+        if conf[k] < thresh:
+            fn += 1
+        elif v != labels[k]:
+            fp += 1
+        else:
+            tp += 1
+        t += 1
+    #print runs[167],labels[167]
+    #print tp,fp,fn
+    #print("Precision: ",tp/(tp+fp))
+    #print("False positives: ",fp/(tp+fp))
+    #print("Recall: ",1.0-fn/t)
+    try:
+        print str(1.0-fn/t)+","+str(tp/(tp+fp))
+    except:
+        print "0,1"