From 1072e90a1917b73b2213d0b76286008e5655d1f5 Mon Sep 17 00:00:00 2001 From: Thibaut Horel Date: Wed, 15 Feb 2012 18:21:38 -0800 Subject: Add brightness.py to compute the brightness of images. --- data/brightness.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 data/brightness.py (limited to 'data') 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) + + -- cgit v1.2.3-70-g09d2