Python is_vision_availableの例

コード例 #1

ファイル: test_pipelines_image_segmentation.py プロジェクト: cahya-wirawan/transformers

    is_vision_available,
    pipeline,
)
from transformers.testing_utils import (
    is_pipeline_test,
    nested_simplify,
    require_tf,
    require_timm,
    require_torch,
    require_vision,
    slow,
)

from .test_pipelines_common import ANY, PipelineTestCaseMeta

if is_vision_available():
    from PIL import Image
else:

    class Image:
        @staticmethod
        def open(*args, **kwargs):
            pass


def hashimage(image: Image) -> str:
    m = hashlib.md5(image.tobytes())
    return m.hexdigest()


@require_vision

コード例 #2

ファイル: test_modeling_detr.py プロジェクト: yulinggu-cs/transformers

 def default_feature_extractor(self):
     return DetrFeatureExtractor.from_pretrained(
         "facebook/detr-resnet-50") if is_vision_available() else None

コード例 #3

ファイル: test_modeling_maskformer.py プロジェクト: yulinggu-cs/transformers

 def default_feature_extractor(self):
     return MaskFormerFeatureExtractor.from_pretrained(
         self.model_checkpoints) if is_vision_available() else None

コード例 #4

ファイル: test_modeling_vilt.py プロジェクト: huggingface/transformers

 def default_processor(self):
     return ViltProcessor.from_pretrained(
         "dandelin/vilt-b32-finetuned-vqa") if is_vision_available(
         ) else None