mindformers.models.swin.SwinProcessor¶

class mindformers.models.swin.SwinProcessor(image_processor=None, return_tensors='ms')[源代码]¶

Swin processor, consists of a feature extractor (BaseFeatureEXtractor) for image input.

Examples:

>>> from mindformers import MindFormerBook
>>> from mindformers.models import SwinProcessor
>>> yaml_path = os.path.join(MindFormerBook.get_project_path(), "configs",
...                          "swin", "run_swin_base_p4w7_224_100ep.yaml")
>>> # build SwinProcessor from pretrained
>>> pro_a = SwinProcessor.from_pretrained('swin_base_p4w7')
>>> # build SwinProcessor from config
>>> pro_b = SwinProcessor.from_pretrained(yaml_path)