def generate(self, namespace, recipe_context={}):
from PYME.IO.rgb_image import image_to_rgb, image_to_cmy
im = namespace[self.inputName]
if self.colorblindFriendly:
return image_to_cmy(im, zoom=self.zoom, scaling=self.scaling, scaling_factor=self.scaling_factor)
else:
return image_to_rgb(im, zoom=self.zoom, scaling=self.scaling, scaling_factor=self.scaling_factor)
def _save(self, image, path):
from PIL import Image
from PYME.IO.rgb_image import image_to_rgb, image_to_cmy
if (self.colorblind_friendly and (image.data.shape[3] != 1)):
im = image_to_cmy(image,
zoom=self.zoom,
scaling=self.scaling,
scaling_factor=self.scaling_factor)
else:
im = image_to_rgb(image,
zoom=self.zoom,
scaling=self.scaling,
scaling_factor=self.scaling_factor)
rgb = Image.fromarray(im, mode='RGB')
rgb.save(path)