a
    d                     @   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_13complete_box_iou complete_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   o/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/functional/detection/ciou.py_ciou_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   _ciou_compute&   s    r   )r   r	   r
   r   r   r   c                 C   s.   t stdtj dt| |||}t||S )aY	  Compute Complete Intersection over Union (`CIOU`_) 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 complete_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],
        ...     ]
        ... )
        >>> complete_intersection_over_union(preds, target)
        tensor(0.5790)

    Example::
        By setting `aggregate=False` the IoU score per prediction and target boxes is returned:

        >>> import torch
        >>> from torchmetrics.functional.detection import complete_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],
        ...     ]
        ... )
        >>> complete_intersection_over_union(preds, target, aggregate=False)
        tensor([[ 0.6883, -0.2072, -0.3352],
                [-0.2217,  0.4881, -0.1913],
                [-0.3971, -0.1543,  0.5606]])

    `z` requires that `torchvision` version 0.13.0 or newer is installed. Please install with `pip install torchvision>=0.13` 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   
 		   