def ex(params):
rgb = RGBExtractor()
rgb.shp = (200, 200)
p = list(params)
p[-1] *= 1.5
return rgb.extract(*p)
def ex(params):
rgb = RGBExtractor()
rgb.shp = (200, 200)
p = list(params)
p[-1] *= 1.5
return rgb.extract(*p)
def main():
ex = RGBExtractor()
ex.shp = (100, 100)
data = filter(hit_rate=0.2)
params = {}
for i, d in enumerate(data):
d1 = list(d)
d2 = list(d)
d2[-1] *= 2
l = int(np.round(d[0]))
d1 = [l] + d1
d2 = [l] + d2
im = np.hstack((ex(*d1), ex(*d2)))
pth = "vet_%4.4i.png" % i
params[pth] = d
pth = os.path.join('best', pth)
imsave(pth, im)
pth = os.path.join('best', 'data.pkl')
with open(pth, 'w') as outfile:
pickle.dump(params, outfile)
def collage(bubbles):
ex = RGBExtractor()
ex.shp = (200, 200)
images = [ex.extract(*p) for p in bubble_params(bubbles)]
if len(images) == 3:
return np.vstack(images)
r, g, b = tuple(
montage2d(np.array([a[:, :, i] for a in images])) for i in range(3))
return np.dstack((r, g, b)).astype(np.uint8)
def collage(bubbles):
ex = RGBExtractor()
ex.shp = (200, 200)
images = [ex.extract(*p) for p in bubble_params(bubbles)]
if len(images) == 3:
return np.vstack(images)
r, g, b = tuple(montage2d(np.array([a[:, :, i] for a in images]))
for i in range(3))
return np.dstack((r, g, b)).astype(np.uint8)
