





import unittest

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

from .test_pipelines_common import ANY


if is_vision_available():
							from PIL import Image
else:
							class     UpperCamelCase__					:






									'''simple docstring'''






									@staticmethod
									def 						snake_case__				(							*snake_case__,   **snake_case__  )       -> Optional[int]:






																"""simple docstring"""
																pass
@is_pipeline_test
@require_vision
@require_torch
class     UpperCamelCase__					(     unittest.TestCase				):






		'''simple docstring'''






		__a       :       Any       =	MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING






		def 						snake_case__				(							self,   snake_case__,   snake_case__,   snake_case__  )       -> Union[str, Any]:






									"""simple docstring"""
									lowercase_	:   int  				=     pipeline(
									    """zero-shot-object-detection""",   model="""hf-internal-testing/tiny-random-owlvit-object-detection"""  )

									lowercase_	:   Optional[int]  				=     [
									    {
									        """image""": """./tests/fixtures/tests_samples/COCO/000000039769.png""",
									        """candidate_labels""": ["""cat""", """remote""", """couch"""],
									    }
									]
									return object_detector, examples






		def 						snake_case__				(							self,   snake_case__,   snake_case__  )       -> List[Any]:






									"""simple docstring"""
									lowercase_	:   Union[str, Any]  				=     object_detector(examples[0],   threshold=0.0  )

									lowercase_	:   List[str]  				=     len(snake_case__  )
									self.assertGreater(snake_case__,   0  )
									self.assertEqual(
									    snake_case__,   [
									        {
									            """score""": ANY(snake_case__  ),
									            """label""": ANY(snake_case__  ),
									            """box""": {"""xmin""": ANY(snake_case__  ), """ymin""": ANY(snake_case__  ), """xmax""": ANY(snake_case__  ), """ymax""": ANY(snake_case__  )},
									        }
									        for i in range(snake_case__  )
									    ],   )






		@require_tf
		@unittest.skip("""Zero Shot Object Detection not implemented in TF"""  )
		def 						snake_case__				(							self  )       -> int:






									"""simple docstring"""
									pass






		@require_torch
		def 						snake_case__				(							self  )       -> List[Any]:






									"""simple docstring"""
									lowercase_	:   List[Any]  				=     pipeline(
									    """zero-shot-object-detection""",   model="""hf-internal-testing/tiny-random-owlvit-object-detection"""  )

									lowercase_	:   Any  				=     object_detector(
									    """./tests/fixtures/tests_samples/COCO/000000039769.png""",   candidate_labels=["""cat""", """remote""", """couch"""],   threshold=0.64,   )

									self.assertEqual(
									    nested_simplify(snake_case__,   decimals=4  ),   [
									        {"""score""": 0.7235, """label""": """cat""", """box""": {"""xmin""": 2_04, """ymin""": 1_67, """xmax""": 2_32, """ymax""": 1_90}},
									        {"""score""": 0.7218, """label""": """remote""", """box""": {"""xmin""": 2_04, """ymin""": 1_67, """xmax""": 2_32, """ymax""": 1_90}},
									        {"""score""": 0.7184, """label""": """couch""", """box""": {"""xmin""": 2_04, """ymin""": 1_67, """xmax""": 2_32, """ymax""": 1_90}},
									        {"""score""": 0.6748, """label""": """remote""", """box""": {"""xmin""": 5_71, """ymin""": 83, """xmax""": 5_98, """ymax""": 1_03}},
									        {"""score""": 0.6656, """label""": """cat""", """box""": {"""xmin""": 5_71, """ymin""": 83, """xmax""": 5_98, """ymax""": 1_03}},
									        {"""score""": 0.6614, """label""": """couch""", """box""": {"""xmin""": 5_71, """ymin""": 83, """xmax""": 5_98, """ymax""": 1_03}},
									        {"""score""": 0.6456, """label""": """remote""", """box""": {"""xmin""": 4_94, """ymin""": 1_05, """xmax""": 5_21, """ymax""": 1_27}},
									        {"""score""": 0.642, """label""": """remote""", """box""": {"""xmin""": 67, """ymin""": 2_74, """xmax""": 93, """ymax""": 2_97}},
									        {"""score""": 0.6419, """label""": """cat""", """box""": {"""xmin""": 4_94, """ymin""": 1_05, """xmax""": 5_21, """ymax""": 1_27}},
									    ],   )

									lowercase_	:   Optional[int]  				=     object_detector(
									    [
									        {
									            """image""": """./tests/fixtures/tests_samples/COCO/000000039769.png""",
									            """candidate_labels""": ["""cat""", """remote""", """couch"""],
									        }
									    ],   threshold=0.64,   )

									self.assertEqual(
									    nested_simplify(snake_case__,   decimals=4  ),   [
									        [
									            {"""score""": 0.7235, """label""": """cat""", """box""": {"""xmin""": 2_04, """ymin""": 1_67, """xmax""": 2_32, """ymax""": 1_90}},
									            {"""score""": 0.7218, """label""": """remote""", """box""": {"""xmin""": 2_04, """ymin""": 1_67, """xmax""": 2_32, """ymax""": 1_90}},
									            {"""score""": 0.7184, """label""": """couch""", """box""": {"""xmin""": 2_04, """ymin""": 1_67, """xmax""": 2_32, """ymax""": 1_90}},
									            {"""score""": 0.6748, """label""": """remote""", """box""": {"""xmin""": 5_71, """ymin""": 83, """xmax""": 5_98, """ymax""": 1_03}},
									            {"""score""": 0.6656, """label""": """cat""", """box""": {"""xmin""": 5_71, """ymin""": 83, """xmax""": 5_98, """ymax""": 1_03}},
									            {"""score""": 0.6614, """label""": """couch""", """box""": {"""xmin""": 5_71, """ymin""": 83, """xmax""": 5_98, """ymax""": 1_03}},
									            {"""score""": 0.6456, """label""": """remote""", """box""": {"""xmin""": 4_94, """ymin""": 1_05, """xmax""": 5_21, """ymax""": 1_27}},
									            {"""score""": 0.642, """label""": """remote""", """box""": {"""xmin""": 67, """ymin""": 2_74, """xmax""": 93, """ymax""": 2_97}},
									            {"""score""": 0.6419, """label""": """cat""", """box""": {"""xmin""": 4_94, """ymin""": 1_05, """xmax""": 5_21, """ymax""": 1_27}},
									        ]
									    ],   )






		@require_torch
		@slow
		def 						snake_case__				(							self  )       -> List[str]:






									"""simple docstring"""
									lowercase_	:   Tuple  				=     pipeline("""zero-shot-object-detection"""  )

									lowercase_	:   List[Any]  				=     object_detector(
									    """http://images.cocodataset.org/val2017/000000039769.jpg""",   candidate_labels=["""cat""", """remote""", """couch"""],   )
									self.assertEqual(
									    nested_simplify(snake_case__,   decimals=4  ),   [
									        {"""score""": 0.2868, """label""": """cat""", """box""": {"""xmin""": 3_24, """ymin""": 20, """xmax""": 6_40, """ymax""": 3_73}},
									        {"""score""": 0.277, """label""": """remote""", """box""": {"""xmin""": 40, """ymin""": 72, """xmax""": 1_77, """ymax""": 1_15}},
									        {"""score""": 0.2537, """label""": """cat""", """box""": {"""xmin""": 1, """ymin""": 55, """xmax""": 3_15, """ymax""": 4_72}},
									        {"""score""": 0.1474, """label""": """remote""", """box""": {"""xmin""": 3_35, """ymin""": 74, """xmax""": 3_71, """ymax""": 1_87}},
									        {"""score""": 0.1208, """label""": """couch""", """box""": {"""xmin""": 4, """ymin""": 0, """xmax""": 6_42, """ymax""": 4_76}},
									    ],   )

									lowercase_	:   Optional[Any]  				=     object_detector(
									    [
									        {
									            """image""": """http://images.cocodataset.org/val2017/000000039769.jpg""",
									            """candidate_labels""": ["""cat""", """remote""", """couch"""],
									        },
									        {
									            """image""": """http://images.cocodataset.org/val2017/000000039769.jpg""",
									            """candidate_labels""": ["""cat""", """remote""", """couch"""],
									        },
									    ],   )
									self.assertEqual(
									    nested_simplify(snake_case__,   decimals=4  ),   [
									        [
									            {"""score""": 0.2868, """label""": """cat""", """box""": {"""xmin""": 3_24, """ymin""": 20, """xmax""": 6_40, """ymax""": 3_73}},
									            {"""score""": 0.277, """label""": """remote""", """box""": {"""xmin""": 40, """ymin""": 72, """xmax""": 1_77, """ymax""": 1_15}},
									            {"""score""": 0.2537, """label""": """cat""", """box""": {"""xmin""": 1, """ymin""": 55, """xmax""": 3_15, """ymax""": 4_72}},
									            {"""score""": 0.1474, """label""": """remote""", """box""": {"""xmin""": 3_35, """ymin""": 74, """xmax""": 3_71, """ymax""": 1_87}},
									            {"""score""": 0.1208, """label""": """couch""", """box""": {"""xmin""": 4, """ymin""": 0, """xmax""": 6_42, """ymax""": 4_76}},
									        ],
									        [
									            {"""score""": 0.2868, """label""": """cat""", """box""": {"""xmin""": 3_24, """ymin""": 20, """xmax""": 6_40, """ymax""": 3_73}},
									            {"""score""": 0.277, """label""": """remote""", """box""": {"""xmin""": 40, """ymin""": 72, """xmax""": 1_77, """ymax""": 1_15}},
									            {"""score""": 0.2537, """label""": """cat""", """box""": {"""xmin""": 1, """ymin""": 55, """xmax""": 3_15, """ymax""": 4_72}},
									            {"""score""": 0.1474, """label""": """remote""", """box""": {"""xmin""": 3_35, """ymin""": 74, """xmax""": 3_71, """ymax""": 1_87}},
									            {"""score""": 0.1208, """label""": """couch""", """box""": {"""xmin""": 4, """ymin""": 0, """xmax""": 6_42, """ymax""": 4_76}},
									        ],
									    ],   )






		@require_tf
		@unittest.skip("""Zero Shot Object Detection not implemented in TF"""  )
		def 						snake_case__				(							self  )       -> Optional[Any]:






									"""simple docstring"""
									pass






		@require_torch
		@slow
		def 						snake_case__				(							self  )       -> Union[str, Any]:






									"""simple docstring"""
									lowercase_	:   Optional[int]  				=     0.2
									lowercase_	:   int  				=     pipeline("""zero-shot-object-detection"""  )

									lowercase_	:   List[Any]  				=     object_detector(
									    """http://images.cocodataset.org/val2017/000000039769.jpg""",   candidate_labels=["""cat""", """remote""", """couch"""],   threshold=snake_case__,   )
									self.assertEqual(
									    nested_simplify(snake_case__,   decimals=4  ),   [
									        {"""score""": 0.2868, """label""": """cat""", """box""": {"""xmin""": 3_24, """ymin""": 20, """xmax""": 6_40, """ymax""": 3_73}},
									        {"""score""": 0.277, """label""": """remote""", """box""": {"""xmin""": 40, """ymin""": 72, """xmax""": 1_77, """ymax""": 1_15}},
									        {"""score""": 0.2537, """label""": """cat""", """box""": {"""xmin""": 1, """ymin""": 55, """xmax""": 3_15, """ymax""": 4_72}},
									    ],   )






		@require_torch
		@slow
		def 						snake_case__				(							self  )       -> Any:






									"""simple docstring"""
									lowercase_	:   Union[str, Any]  				=     2
									lowercase_	:   str  				=     pipeline("""zero-shot-object-detection"""  )

									lowercase_	:   Tuple  				=     object_detector(
									    """http://images.cocodataset.org/val2017/000000039769.jpg""",   candidate_labels=["""cat""", """remote""", """couch"""],   top_k=snake_case__,   )
									self.assertEqual(
									    nested_simplify(snake_case__,   decimals=4  ),   [
									        {"""score""": 0.2868, """label""": """cat""", """box""": {"""xmin""": 3_24, """ymin""": 20, """xmax""": 6_40, """ymax""": 3_73}},
									        {"""score""": 0.277, """label""": """remote""", """box""": {"""xmin""": 40, """ymin""": 72, """xmax""": 1_77, """ymax""": 1_15}},
									    ],   )






from typing import Dict, List, Optional, Union

import numpy as np

from transformers.utils import is_vision_available
from transformers.utils.generic import TensorType

from ...image_processing_utils import BaseImageProcessor, BatchFeature, get_size_dict
from ...image_transforms import (
    center_crop,
    get_resize_output_image_size,
    normalize,
    rescale,
    resize,
    to_channel_dimension_format,
)
from ...image_utils import (
    IMAGENET_STANDARD_MEAN,
    IMAGENET_STANDARD_STD,
    ChannelDimension,
    ImageInput,
    PILImageResampling,
    is_valid_image,
    to_numpy_array,
    valid_images,
)
from ...utils import logging


if is_vision_available():
							import PIL

UpperCAmelCase_							=				logging.get_logger(__name__)





def  __magic_name__  (				lowercase				)				-> List[List[ImageInput]]:

							"""simple docstring"""



							if isinstance(lowercase				,       (list, tuple)				) and isinstance(videos[0]				,       (list, tuple)				) and is_valid_image(videos[0][0]				):
														return videos

							elif isinstance(lowercase				,       (list, tuple)				) and is_valid_image(videos[0]				):
														return [videos]

							elif is_valid_image(lowercase				):
														return [[videos]]

							raise ValueError(f"""Could not make batched video from {videos}"""				)




class     UpperCamelCase__					(     lowerCamelCase__				):






		'''simple docstring'''






		__a       :       List[Any]       =	["""pixel_values"""]






		def __init__(							self,   snake_case__ = True,   snake_case__ = None,   snake_case__ = PILImageResampling.BILINEAR,   snake_case__ = True,   snake_case__ = None,   snake_case__ = True,   snake_case__ = 1 / 2_55,   snake_case__ = True,   snake_case__ = True,   snake_case__ = None,   snake_case__ = None,   **snake_case__,   )       -> None:






									"""simple docstring"""
									super().__init__(**snake_case__  )
									lowercase_	:   str  				=     size if size is not None else {"""shortest_edge""": 2_56}
									lowercase_	:   int  				=     get_size_dict(snake_case__,   default_to_square=snake_case__  )
									lowercase_	:   int  				=     crop_size if crop_size is not None else {"""height""": 2_24, """width""": 2_24}
									lowercase_	:   Optional[int]  				=     get_size_dict(snake_case__,   param_name="""crop_size"""  )

									lowercase_	:   List[Any]  				=     do_resize
									lowercase_	:   int  				=     size
									lowercase_	:   int  				=     do_center_crop
									lowercase_	:   Optional[Any]  				=     crop_size
									lowercase_	:   str  				=     resample
									lowercase_	:   Any  				=     do_rescale
									lowercase_	:   Dict  				=     rescale_factor
									lowercase_	:   List[Any]  				=     offset
									lowercase_	:   int  				=     do_normalize
									lowercase_	:   str  				=     image_mean if image_mean is not None else IMAGENET_STANDARD_MEAN
									lowercase_	:   Tuple  				=     image_std if image_std is not None else IMAGENET_STANDARD_STD






		def 						snake_case__				(							self,   snake_case__,   snake_case__,   snake_case__ = PILImageResampling.BILINEAR,   snake_case__ = None,   **snake_case__,   )       -> np.ndarray:






									"""simple docstring"""
									lowercase_	:   int  				=     get_size_dict(snake_case__,   default_to_square=snake_case__  )
									if "shortest_edge" in size:
																lowercase_	:   Union[str, Any]  				=     get_resize_output_image_size(snake_case__,   size["""shortest_edge"""],   default_to_square=snake_case__  )
									elif "height" in size and "width" in size:
																lowercase_	:   Dict  				=     (size["""height"""], size["""width"""])
									else:
																raise ValueError(f"""Size must have 'height' and 'width' or 'shortest_edge' as keys. Got {size.keys()}"""  )
									return resize(snake_case__,   size=snake_case__,   resample=snake_case__,   data_format=snake_case__,   **snake_case__  )






		def 						snake_case__				(							self,   snake_case__,   snake_case__,   snake_case__ = None,   **snake_case__,   )       -> np.ndarray:






									"""simple docstring"""
									lowercase_	:   List[Any]  				=     get_size_dict(snake_case__  )
									if "height" not in size or "width" not in size:
																raise ValueError(f"""Size must have 'height' and 'width' as keys. Got {size.keys()}"""  )
									return center_crop(snake_case__,   size=(size["""height"""], size["""width"""]),   data_format=snake_case__,   **snake_case__  )






		def 						snake_case__				(							self,   snake_case__,   snake_case__,   snake_case__ = True,   snake_case__ = None,   **snake_case__,   )       -> str:






									"""simple docstring"""
									lowercase_	:   Dict  				=     image.astype(np.floataa  )
									if offset:
																lowercase_	:   Optional[Any]  				=     image - (scale / 2)
									return rescale(snake_case__,   scale=snake_case__,   data_format=snake_case__,   **snake_case__  )






		def 						snake_case__				(							self,   snake_case__,   snake_case__,   snake_case__,   snake_case__ = None,   **snake_case__,   )       -> np.ndarray:






									"""simple docstring"""
									return normalize(snake_case__,   mean=snake_case__,   std=snake_case__,   data_format=snake_case__,   **snake_case__  )






		def 						snake_case__				(							self,   snake_case__,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = ChannelDimension.FIRST,   )       -> np.ndarray:






									"""simple docstring"""
									if do_resize and size is None or resample is None:
																raise ValueError("""Size and resample must be specified if do_resize is True."""  )

									if do_center_crop and crop_size is None:
																raise ValueError("""Crop size must be specified if do_center_crop is True."""  )

									if do_rescale and rescale_factor is None:
																raise ValueError("""Rescale factor must be specified if do_rescale is True."""  )

									if do_normalize and (image_mean is None or image_std is None):
																raise ValueError("""Image mean and std must be specified if do_normalize is True."""  )

									if offset and not do_rescale:
																raise ValueError("""For offset, do_rescale must also be set to True."""  )

									# All transformations expect numpy arrays.
									lowercase_	:   Optional[int]  				=     to_numpy_array(snake_case__  )

									if do_resize:
																lowercase_	:   Dict  				=     self.resize(image=snake_case__,   size=snake_case__,   resample=snake_case__  )

									if do_center_crop:
																lowercase_	:   Optional[int]  				=     self.center_crop(snake_case__,   size=snake_case__  )

									if do_rescale:
																lowercase_	:   Dict  				=     self.rescale(image=snake_case__,   scale=snake_case__,   offset=snake_case__  )

									if do_normalize:
																lowercase_	:   Optional[Any]  				=     self.normalize(image=snake_case__,   mean=snake_case__,   std=snake_case__  )

									lowercase_	:   Tuple  				=     to_channel_dimension_format(snake_case__,   snake_case__  )
									return image






		def 						snake_case__				(							self,   snake_case__,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = None,   snake_case__ = ChannelDimension.FIRST,   **snake_case__,   )       -> PIL.Image.Image:






									"""simple docstring"""
									lowercase_	:   Any  				=     do_resize if do_resize is not None else self.do_resize
									lowercase_	:   str  				=     resample if resample is not None else self.resample
									lowercase_	:   List[str]  				=     do_center_crop if do_center_crop is not None else self.do_center_crop
									lowercase_	:   Optional[Any]  				=     do_rescale if do_rescale is not None else self.do_rescale
									lowercase_	:   str  				=     rescale_factor if rescale_factor is not None else self.rescale_factor
									lowercase_	:   List[Any]  				=     offset if offset is not None else self.offset
									lowercase_	:   Optional[Any]  				=     do_normalize if do_normalize is not None else self.do_normalize
									lowercase_	:   Any  				=     image_mean if image_mean is not None else self.image_mean
									lowercase_	:   Optional[Any]  				=     image_std if image_std is not None else self.image_std

									lowercase_	:   Optional[Any]  				=     size if size is not None else self.size
									lowercase_	:   Dict  				=     get_size_dict(snake_case__,   default_to_square=snake_case__  )
									lowercase_	:   Union[str, Any]  				=     crop_size if crop_size is not None else self.crop_size
									lowercase_	:   Tuple  				=     get_size_dict(snake_case__,   param_name="""crop_size"""  )

									if not valid_images(snake_case__  ):
																raise ValueError(
																    """Invalid image type. Must be of type PIL.Image.Image, numpy.ndarray, """
																    """torch.Tensor, tf.Tensor or jax.ndarray."""  )

									lowercase_	:   Optional[Any]  				=     make_batched(snake_case__  )

									lowercase_	:   Optional[Any]  				=     [
									    [
									        self._preprocess_image(
									            image=snake_case__,   do_resize=snake_case__,   size=snake_case__,   resample=snake_case__,   do_center_crop=snake_case__,   crop_size=snake_case__,   do_rescale=snake_case__,   rescale_factor=snake_case__,   offset=snake_case__,   do_normalize=snake_case__,   image_mean=snake_case__,   image_std=snake_case__,   data_format=snake_case__,   )
									        for img in video
									    ]
									    for video in videos
									]

									lowercase_	:   List[str]  				=     {"""pixel_values""": videos}
									return BatchFeature(data=snake_case__,   tensor_type=snake_case__  )