





# Copyright 2023 The HuggingFace Team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from typing import TYPE_CHECKING

# rely on isort to merge the imports
from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available


snake_case_           =      {
    'configuration_vivit': ['VIVIT_PRETRAINED_CONFIG_ARCHIVE_MAP', 'VivitConfig'],
}
try:
				if not is_vision_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				snake_case_           =      ['VivitImageProcessor']


try:
				if not is_torch_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				snake_case_           =      [
				    'VIVIT_PRETRAINED_MODEL_ARCHIVE_LIST',
				    'VivitModel',
				    'VivitPreTrainedModel',
				    'VivitForVideoClassification',
				]


if TYPE_CHECKING:
				from .configuration_vivit import VIVIT_PRETRAINED_CONFIG_ARCHIVE_MAP, VivitConfig

				try:
								if not is_vision_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .image_processing_vivit import VivitImageProcessor

				try:
								if not is_torch_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .modeling_vivit import (
								    VIVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
								    VivitForVideoClassification,
								    VivitModel,
								    VivitPreTrainedModel,
								)


else:
				import sys

				snake_case_           =      _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)








from typing import List, Union

import numpy as np

from ..utils import add_end_docstrings, is_torch_available, is_vision_available, logging, requires_backends
from .base import PIPELINE_INIT_ARGS, Pipeline


if is_vision_available():
    from PIL import Image

    from ..image_utils import load_image

if is_torch_available():
    import torch

    from ..models.auto.modeling_auto import MODEL_FOR_DEPTH_ESTIMATION_MAPPING

snake_case_           =      logging.get_logger(__name__)

@add_end_docstrings(_UpperCAmelCase  )
class 		SCREAMING_SNAKE_CASE__    (	_UpperCAmelCase  ):


      def __init__(self	:				int    ,      *a__	:				List[Any]    ,      **a__	:				Dict				):





         """simple docstring"""



         super().__init__(*a__    ,      **a__				)
         requires_backends(self    ,      '''vision'''				)
         self.check_model_type(a__				)


      def __call__(self	:				Optional[Any]    ,      a__	:				Union[str, List[str], "Image.Image", List["Image.Image"]]    ,      **a__	:				List[str]				):





         """simple docstring"""



         return super().__call__(a__    ,      **a__				)


      def        a			(self	:				int    ,      **a__	:				int				):





         """simple docstring"""



         return {}, {}, {}


      def        a			(self	:				Optional[int]    ,      a__	:				Optional[int]				):





         """simple docstring"""



         __snake_case								=       load_image(a__				)
         __snake_case								=       image.size
         __snake_case								=       self.image_processor(images=a__    ,      return_tensors=self.framework				)
         return model_inputs


      def        a			(self	:				List[Any]    ,      a__	:				Union[str, Any]				):





         """simple docstring"""



         __snake_case								=       self.model(**a__				)
         return model_outputs


      def        a			(self	:				int    ,      a__	:				str				):





         """simple docstring"""



         __snake_case								=       model_outputs.predicted_depth
         __snake_case								=       torch.nn.functional.interpolate(
             predicted_depth.unsqueeze(1				)    ,      size=self.image_size[::-1]    ,      mode='''bicubic'''    ,      align_corners=a__				)
         __snake_case								=       prediction.squeeze().cpu().numpy()
         __snake_case								=       (output * 255 / np.max(a__				)).astype('''uint8'''				)
         __snake_case								=       Image.fromarray(a__				)
         __snake_case								=       {}
         __snake_case								=       predicted_depth
         __snake_case								=       depth
         return output_dict

