Example #1
0
    def _pre_process_image(self, img: PIL):
        img = img.resize((self.run_parameters.get('input_shape')[0],
                          self.run_parameters.get('input_shape')[1]),
                         Image.ANTIALIAS)
        if self.run_parameters.get('greyscale'):
            img = img.convert.convert('LA')
        else:
            img = img.convert('RGB')
        img = np.array(img)
        img = img / 255
        # casts list to tuple
        img_shape = (1, ) + tuple(self.run_parameters.get('input_shape'))

        img = img.reshape(img_shape)
        return img