o
    .wÖi7	  ã                   @   sX   d dl mZ d dlmZ d dlmZmZ d dlmZ G dd„ deƒZ	G dd„ deƒZ
d	S )
é    )Ú
Collection)ÚAny)ÚModifiedPanopticQualityÚPanopticQuality)Ú_deprecated_root_import_classc                       óB   e Zd ZdZ	ddee dee dededdf
‡ fd	d
„Z‡  Z	S )Ú_ModifiedPanopticQualitya{  Wrapper for deprecated import.

    >>> from torch import tensor
    >>> preds = tensor([[[0, 0], [0, 1], [6, 0], [7, 0], [0, 2], [1, 0]]])
    >>> target = tensor([[[0, 1], [0, 0], [6, 0], [7, 0], [6, 0], [255, 0]]])
    >>> pq_modified = _ModifiedPanopticQuality(things = {0, 1}, stuffs = {6, 7})
    >>> pq_modified(preds, target)
    tensor(0.7667, dtype=torch.float64)

    FÚthingsÚstuffsÚallow_unknown_preds_categoryÚkwargsÚreturnNc                    ó(   t ddƒ tƒ jd|||dœ|¤Ž d S )Nr   Ú	detection©r	   r
   r   © ©r   ÚsuperÚ__init__©Úselfr	   r
   r   r   ©Ú	__class__r   ú_/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/torchmetrics/detection/_deprecated.pyr      ó   
ÿ
ÿz!_ModifiedPanopticQuality.__init__©F©
Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚintÚboolr   r   Ú__classcell__r   r   r   r   r      s    üþýüûúr   c                       r   )Ú_PanopticQualitya;  Wrapper for deprecated import.

    >>> from torch import tensor
    >>> preds = tensor([[[[6, 0], [0, 0], [6, 0], [6, 0]],
    ...                  [[0, 0], [0, 0], [6, 0], [0, 1]],
    ...                  [[0, 0], [0, 0], [6, 0], [0, 1]],
    ...                  [[0, 0], [7, 0], [6, 0], [1, 0]],
    ...                  [[0, 0], [7, 0], [7, 0], [7, 0]]]])
    >>> target = tensor([[[[6, 0], [0, 1], [6, 0], [0, 1]],
    ...                   [[0, 1], [0, 1], [6, 0], [0, 1]],
    ...                   [[0, 1], [0, 1], [6, 0], [1, 0]],
    ...                   [[0, 1], [7, 0], [1, 0], [1, 0]],
    ...                   [[0, 1], [7, 0], [7, 0], [7, 0]]]])
    >>> panoptic_quality = _PanopticQuality(things = {0, 1}, stuffs = {6, 7})
    >>> panoptic_quality(preds, target)
    tensor(0.5463, dtype=torch.float64)

    Fr	   r
   r   r   r   Nc                    r   )Nr   r   r   r   r   r   r   r   r   r   5   r   z_PanopticQuality.__init__r   r   r   r   r   r   r$   !   s    üþýüûúr$   N)Úcollections.abcr   Útypingr   Útorchmetrics.detectionr   r   Útorchmetrics.utilities.printsr   r   r$   r   r   r   r   Ú<module>   s    