def transform_image(image, params=[], tags=[]):
image = apply_tags(image, tags)
image = extract_roi(image, params)
for k, v in params:
if k == 'saturation':
saturation = float(v)
image = scale_saturation(image, saturation)
elif k == 'gamma':
gamma = float(v)
image = np.power(img_as_float(image), gamma)
elif k == 'brightness':
b = float(v)
image = (img_as_float(image) * b).clip(0., 1.)
elif k == 'width':
scale = float(v) / image.shape[1]
image = rescale(image, scale)
elif k == 'maxwidth':
w = float(v)
if image.shape[1] > w:
scale = w / image.shape[1]
image = rescale(image, scale)
elif k == 'height':
scale = float(v) / image.shape[0]
image = rescale(image, scale)
elif k == 'maxheight':
h = float(v)
if image.shape[0] > h:
scale = h / image.shape[0]
image = rescale(image, scale)
return image
def transform_image(image, params=[], tags=[]):
image = apply_tags(image, tags)
image = extract_roi(image, params)
for k,v in params:
if k=='saturation':
saturation = float(v)
image = scale_saturation(image, saturation)
elif k=='gamma':
gamma = float(v)
image = np.power(img_as_float(image), gamma)
elif k=='brightness':
b = float(v)
image = (img_as_float(image) * b).clip(0.,1.)
elif k=='width':
scale = float(v) / image.shape[1]
image = rescale(image, scale)
elif k=='maxwidth':
w = float(v)
if image.shape[1] > w:
scale = w / image.shape[1]
image = rescale(image, scale)
elif k=='height':
scale = float(v) / image.shape[0]
image = rescale(image, scale)
elif k=='maxheight':
h = float(v)
if image.shape[0] > h:
scale = h / image.shape[0]
image = rescale(image, scale)
return image
def show_image(image,max_width=None,order=1):
f = tmp_img()
if image.dtype == np.bool:
image = image.astype(np.int) * 255
if max_width is not None:
(h,w) = image.shape[:2]
scale = max_width / float(w)
image = rescale(image,scale,order=order)
imsave(f,image)
d = display.Image(filename=f)
os.remove(f)
return d
