In [5]:
from os import listdir
from os.path import isfile, join
mypath='../../../antarctica/flags/flat/'
onlyfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]
In [8]:
from scipy.misc import imread
In [11]:
im=imread(mypath+onlyfiles[0],mode='RGB')
In [12]:
import matplotlib.pyplot as plt
%matplotlib inline
In [13]:
plt.imshow(im)
Out[13]:
In [15]:
im=imread(mypath+onlyfiles[30],mode='RGB')
plt.imshow(im)
Out[15]:
In [27]:
import numpy as np
In [28]:
r=np.array(im.T[0]).flatten()
g=np.array(im.T[1]).flatten()
b=np.array(im.T[2]).flatten()
In [34]:
R=sum(r)*1.0/(len(r)*256)
G=sum(g)*1.0/(len(g)*256)
B=sum(b)*1.0/(len(b)*256)
In [37]:
R,G,B
Out[37]:
In [68]:
im=imread(mypath+onlyfiles[145],mode='RGB')
plt.imshow(im)
r=np.array(im.T[0]).flatten()
g=np.array(im.T[1]).flatten()
b=np.array(im.T[2]).flatten()
R=sum(r)*1.0/(len(r)*256)
G=sum(g)*1.0/(len(g)*256)
B=sum(b)*1.0/(len(b)*256)
R,G,B
Out[68]:
In [69]:
import json
In [75]:
data={}
for i in range(len(onlyfiles)):
im=imread(mypath+onlyfiles[i],mode='RGB')
plt.imshow(im)
r=np.array(im.T[0]).flatten()
g=np.array(im.T[1]).flatten()
b=np.array(im.T[2]).flatten()
R=sum(r)*1.0/(len(r)*256)
G=sum(g)*1.0/(len(g)*256)
B=sum(b)*1.0/(len(b)*256)
data[onlyfiles[i][:2]]=[R,G,B,onlyfiles[i][:2]]
In [76]:
file('data.json','w').write(json.dumps(data))