
    Ph                         d dl Z ddlmZmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ e j                  j                  d	        Zy)
    N   )BoundingBoxesBoundingBoxFormat)Image)Mask)set_return_type)TVTensor)Videoc                    t        |t              rLt        j                  | |j                  d|j                        |j                  d|j
                              S | j                  t        |            S )a  Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``.

    If ``like`` is a :class:`~torchvision.tv_tensors.BoundingBoxes`, the ``format`` and ``canvas_size`` of
    ``like`` are assigned to ``wrappee``, unless they are passed as ``kwargs``.

    Args:
        wrappee (Tensor): The tensor to convert.
        like (:class:`~torchvision.tv_tensors.TVTensor`): The reference.
            ``wrappee`` will be converted into the same subclass as ``like``.
        kwargs: Can contain "format" and "canvas_size" if ``like`` is a :class:`~torchvision.tv_tensor.BoundingBoxes`.
            Ignored otherwise.
    formatcanvas_size)r   r   )
isinstancer   _wrapgetr   r   as_subclasstype)wrappeelikekwargss      jC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/tv_tensors/__init__.pywrapr      sa     $&""::h4

=$2B2BC
 	
 ""4:..    )torch_bounding_boxesr   r   _imager   _maskr   _torch_function_helpersr   
_tv_tensorr	   _videor
   compilerdisabler    r   r   <module>r#      s8     =   4    / /r   