diff options
Diffstat (limited to 'data/face-frame-recognition-accuracy.py')
| -rwxr-xr-x | data/face-frame-recognition-accuracy.py | 48 |
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" |