1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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)
|
