def preprocess_image()

in src_batch_job/source/model.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

    def preprocess_image(self, image):
        '''
        Preprocess an input Pillow image object.
        '''
        image_nd = self.correct_channel(nd.array(image))
        target_shape = (224, 244)
        resized = mx.image.resize_short(image_nd,
                                        target_shape[0]).astype('float32')
        cropped, crop_info = mx.image.center_crop(resized, target_shape)
        cropped /= 255.
        normalized = mx.image.color_normalize(cropped,
                                              mean=self.MEAN_IMAGE,
                                              std=self.STD_IMAGE)
        transposed = nd.transpose(normalized, (2, 0, 1))
        return transposed