a
    d6                     @   s   d dl mZ d dlZd dlmZmZ er:er:d dlmZ n
dZdgZddgiZ	dej
ej
ee eej
dd	d
Zdej
eej
dddZdej
ej
ee eeej
dddZdS )    )OptionalN)_TORCHVISION_AVAILABLE_TORCHVISION_GREATER_EQUAL_0_8box_iouintersection_over_union)r   Ztorchvision)predstargetiou_thresholdreplacement_valreturnc                 C   s"   t | |}|d ur||||k < |S )Nr   )r   r	   r
   r   iou r   n/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/functional/detection/iou.py_iou_update   s    
r   T)r   	aggregater   c                 C   s0   |s| S |   dkr |   S tjd| jdS )Nr   g        )device)ZnumelZdiagmeantorchZtensorr   )r   r   r   r   r   _iou_compute&   s    r   )r   r	   r
   r   r   r   c                 C   s.   t stdtj dt| |||}t||S )a  Compute Intersection over Union between two sets of boxes.

    Both sets of boxes are expected to be in (x1, y1, x2, y2) format with 0 <= x1 < x2 and 0 <= y1 < y2.

    Args:
        preds:
            The input tensor containing the predicted bounding boxes.
        target:
            The tensor containing the ground truth.
        iou_threshold:
            Optional IoU thresholds for evaluation. If set to `None` the threshold is ignored.
        replacement_val:
            Value to replace values under the threshold with.
        aggregate:
            Return the average value instead of the full matrix of values

    Example::
        By default iou is aggregated across all box pairs e.g. mean along the diagonal of the IoU matrix:

        >>> import torch
        >>> from torchmetrics.functional.detection import intersection_over_union
        >>> preds = torch.tensor(
        ...     [
        ...         [296.55, 93.96, 314.97, 152.79],
        ...         [328.94, 97.05, 342.49, 122.98],
        ...         [356.62, 95.47, 372.33, 147.55],
        ...     ]
        ... )
        >>> target = torch.tensor(
        ...     [
        ...         [300.00, 100.00, 315.00, 150.00],
        ...         [330.00, 100.00, 350.00, 125.00],
        ...         [350.00, 100.00, 375.00, 150.00],
        ...     ]
        ... )
        >>> intersection_over_union(preds, target)
        tensor(0.5879)

    Example::
        By setting `aggregate=False` the full IoU matrix is returned:

        >>> import torch
        >>> from torchmetrics.functional.detection import intersection_over_union
        >>> preds = torch.tensor(
        ...     [
        ...         [296.55, 93.96, 314.97, 152.79],
        ...         [328.94, 97.05, 342.49, 122.98],
        ...         [356.62, 95.47, 372.33, 147.55],
        ...     ]
        ... )
        >>> target = torch.tensor(
        ...     [
        ...         [300.00, 100.00, 315.00, 150.00],
        ...         [330.00, 100.00, 350.00, 125.00],
        ...         [350.00, 100.00, 375.00, 150.00],
        ...     ]
        ... )
        >>> intersection_over_union(preds, target, aggregate=False)
        tensor([[0.6898, 0.0000, 0.0000],
                [0.0000, 0.5086, 0.0000],
                [0.0000, 0.0000, 0.5654]])

    `z` requires that `torchvision` version 0.8.0 or newer is installed. Please install with `pip install torchvision>=0.8` or `pip install torchmetrics[detection]`.)r   ModuleNotFoundErrorr   __name__r   r   )r   r	   r
   r   r   r   r   r   r   r   ,   s    F)r   )T)Nr   T)typingr   r   Ztorchmetrics.utilities.importsr   r   Ztorchvision.opsr   Z__doctest_skip__Z__doctest_requires__ZTensorfloatr   boolr   r   r   r   r   r   <module>   s0   
 		   