In [1]:
from os import listdir
from os.path import isfile, join
mypath='logos/'
onlyfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]
In [2]:
from scipy.misc import imread
In [3]:
im=imread(mypath+onlyfiles[0],mode='RGB')
In [4]:
import matplotlib.pyplot as plt
%matplotlib inline
In [5]:
plt.imshow(im)
Out[5]:
In [6]:
im=imread(mypath+onlyfiles[30],mode='RGB')
plt.imshow(im)
Out[6]:
In [7]:
import numpy as np
In [8]:
r=np.array(im.T[0]).flatten()
g=np.array(im.T[1]).flatten()
b=np.array(im.T[2]).flatten()
In [9]:
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 [10]:
R,G,B
Out[10]:
In [11]:
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[11]:
In [12]:
import json
In [16]:
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][:3]]=[R,G,B,onlyfiles[i][:3]]
In [17]:
file('data.json','w').write(json.dumps(data))