Add brightness.py to compute the brightness of images.

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)
+ 
+