def to_vector()

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

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

    def to_vector(self, image):
        '''
        Vectorize an input Pillow image object.
        '''
        image_t = self.preprocess_image(image)
        output = self._net(image_t.expand_dims(axis=0).as_in_context(self._ctx))
        return output.asnumpy().reshape(-1, )