diff options
| -rw-r--r-- | data/brightness.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/data/brightness.py b/data/brightness.py new file mode 100644 index 0000000..06ddf49 --- /dev/null +++ b/data/brightness.py @@ -0,0 +1,34 @@ +#!/usr/bin/python +import os +import sys +import cv + +numbins = 255 +runs = [] +for arg in sys.argv[2:]: + runs += [[open(arg), sys.argv[1].rstrip('/') + '/' + arg.split('/')[-2] + '/']] + +print "#name,run,frame,brightness" +for lines, pic_dir in runs: + for line in lines: + line = line.strip().split(',') + out = line[0:4] + name,run,frame = out + if frame != 'frame': + filename = pic_dir + frame + '.jpg' + image = cv.LoadImage(filename,cv.CV_LOAD_IMAGE_GRAYSCALE) + hist = cv.CreateHist([numbins+1],cv.CV_HIST_ARRAY,[[0,255]],1) + cv.CalcHist([image],hist) + + #compute brightness + num = 0. + brightness = 0. + for i in range(numbins+1): + temp = cv.QueryHistValue_1D(hist,i) + num += temp + brightness += temp*i + out += [brightness/num] + + print ','.join(out) + + |