mindformers.models.BaseImageProcessor

class mindformers.models.BaseImageProcessor(**kwargs)

BaseImageProcessor for all image preprocess.

实际案例

>>> from mindspore.dataset.vision.transforms import CenterCrop
>>> from mindformers.models.base_processor import BaseImageProcessor
>>> image_resolution = 224
>>> class MyImageProcessor(BaseImageProcessor):
...     def __init__(self, image_resolution):
...         super(MyImageProcessor, self).__init__(image_resolution=image_resolution)
...         self.center_crop = CenterCrop(image_resolution)
...
...     def preprocess(self, images, **kwargs):
...         res = []
...         for image in images:
...             image = self.center_crop(image)
...             res.append(image)
...         return res
...
>>> my_image_processor = MyImageProcessor(image_resolution)
>>> output = my_image_processor(image)

preprocess(images, **kwargs)

preprocess method