o .wi@sddlmZddlmZmZmZddlmZddlm Z ddl m Z m Z ddl mZmZddlmZmZes;dd gZnes@d gZGd dde Zd S) )Sequence)AnyOptionalUnion)Tensor)IntersectionOverUnion) _ciou_compute _ciou_update)_MATPLOTLIB_AVAILABLE_TORCHVISION_AVAILABLE)_AX_TYPE_PLOT_OUT_TYPECompleteIntersectionOverUnion"CompleteIntersectionOverUnion.plotc seZdZUdZdZeed<dZeeed<dZ eed<dZ e ed<d Z e ed <  dd e dee dedededd f fdd ZedededefddZedededefddZ ddeeeeefdeedefddZZS) ra Computes Complete Intersection Over Union (`CIoU`_). As input to ``forward`` and ``update`` the metric accepts the following input: - ``preds`` (:class:`~List`): A list consisting of dictionaries each containing the key-values (each dictionary corresponds to a single image). Parameters that should be provided per dict: - ``boxes`` (:class:`~torch.Tensor`): float tensor of shape ``(num_boxes, 4)`` containing ``num_boxes`` detection boxes of the format specified in the constructor. By default, this method expects ``(xmin, ymin, xmax, ymax)`` in absolute image coordinates. - ``labels`` (:class:`~torch.Tensor`): integer tensor of shape ``(num_boxes)`` containing 0-indexed detection classes for the boxes. - ``target`` (:class:`~List`): A list consisting of dictionaries each containing the key-values (each dictionary corresponds to a single image). Parameters that should be provided per dict: - ``boxes`` (:class:`~torch.Tensor`): float tensor of shape ``(num_boxes, 4)`` containing ``num_boxes`` ground truth boxes of the format specified in the constructor. By default, this method expects ``(xmin, ymin, xmax, ymax)`` in absolute image coordinates. - ``labels`` (:class:`~torch.Tensor`): integer tensor of shape ``(num_boxes)`` containing 0-indexed detection classes for the boxes. As output of ``forward`` and ``compute`` the metric returns the following output: - ``ciou_dict``: A dictionary containing the following key-values: - ciou: (:class:`~torch.Tensor`) with overall ciou value over all classes and samples. - ciou/cl_{cl}: (:class:`~torch.Tensor`), if argument ``class_metrics=True`` Args: box_format: Input format of given boxes. Supported formats are ``[`xyxy`, `xywh`, `cxcywh`]``. iou_thresholds: Optional IoU thresholds for evaluation. If set to `None` the threshold is ignored. class_metrics: Option to enable per-class metrics for IoU. Has a performance impact. respect_labels: Ignore values from boxes that do not have the same label as the ground truth box. Else will compute Iou between all pairs of boxes. kwargs: Additional keyword arguments, see :ref:`Metric kwargs` for more info. Example: >>> import torch >>> from torchmetrics.detection import CompleteIntersectionOverUnion >>> preds = [ ... { ... "boxes": torch.tensor([[296.55, 93.96, 314.97, 152.79], [298.55, 98.96, 314.97, 151.79]]), ... "scores": torch.tensor([0.236, 0.56]), ... "labels": torch.tensor([4, 5]), ... } ... ] >>> target = [ ... { ... "boxes": torch.tensor([[300.00, 100.00, 315.00, 150.00]]), ... "labels": torch.tensor([5]), ... } ... ] >>> metric = CompleteIntersectionOverUnion() >>> metric(preds, target) {'ciou': tensor(0.8611)} Raises: ModuleNotFoundError: If torchvision is not installed with version 0.13.0 or newer. Fis_differentiableThigher_is_betterfull_state_updateciou _iou_typeg _invalid_valxyxyN box_format iou_threshold class_metricsrespect_labelskwargsreturnc s8ts td|jdtj||||fi|dS)NzMetric `zf` requires that `torchvision` is installed. Please install with `pip install torchmetrics[detection]`.)r ModuleNotFoundErrorruppersuper__init__)selfrrrrr __class__X/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/torchmetrics/detection/ciou.pyr js z&CompleteIntersectionOverUnion.__init__argscOt|i|SN)r r&rr$r$r%_iou_update_fnyz,CompleteIntersectionOverUnion._iou_update_fncOr'r()rr)r$r$r%_iou_compute_fn}r+z-CompleteIntersectionOverUnion._iou_compute_fnvalaxcCs |||S)az Plot a single or multiple values from the metric. Args: val: Either a single result from calling `metric.forward` or `metric.compute` or a list of these results. If no value is provided, will automatically call `metric.compute` and plot that result. ax: An matplotlib axis object. If provided will add plot to that axis Returns: Figure object and Axes object Raises: ModuleNotFoundError: If `matplotlib` is not installed .. plot:: :scale: 75 >>> # Example plotting single value >>> import torch >>> from torchmetrics.detection import CompleteIntersectionOverUnion >>> preds = [ ... { ... "boxes": torch.tensor([[296.55, 93.96, 314.97, 152.79], [298.55, 98.96, 314.97, 151.79]]), ... "scores": torch.tensor([0.236, 0.56]), ... "labels": torch.tensor([4, 5]), ... } ... ] >>> target = [ ... { ... "boxes": torch.tensor([[300.00, 100.00, 315.00, 150.00]]), ... "labels": torch.tensor([5]), ... } ... ] >>> metric = CompleteIntersectionOverUnion() >>> metric.update(preds, target) >>> fig_, ax_ = metric.plot() .. plot:: :scale: 75 >>> # Example plotting multiple values >>> import torch >>> from torchmetrics.detection import CompleteIntersectionOverUnion >>> preds = [ ... { ... "boxes": torch.tensor([[296.55, 93.96, 314.97, 152.79], [298.55, 98.96, 314.97, 151.79]]), ... "scores": torch.tensor([0.236, 0.56]), ... "labels": torch.tensor([4, 5]), ... } ... ] >>> target = lambda : [ ... { ... "boxes": torch.tensor([[300.00, 100.00, 315.00, 150.00]]) + torch.randint(-10, 10, (1, 4)), ... "labels": torch.tensor([5]), ... } ... ] >>> metric = CompleteIntersectionOverUnion() >>> vals = [] >>> for _ in range(20): ... vals.append(metric(preds, target())) >>> fig_, ax_ = metric.plot(vals) )_plot)r!r-r.r$r$r%plots Br)rNFT)NN)__name__ __module__ __qualname____doc__rbool__annotations__rrrrstrrfloatrr staticmethodrr*r,rrr r r0 __classcell__r$r$r"r%rsH  D   N)collections.abcrtypingrrrtorchrtorchmetrics.detection.iour&torchmetrics.functional.detection.ciourr torchmetrics.utilities.importsr r torchmetrics.utilities.plotr r __doctest_skip__rr$r$r$r%s