Source code for draugr.torch_utilities.operations.boxes_2d

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "Christian Heider Nielsen"
__doc__ = r"""

           Created on 25/03/2020
           """

import torch

__all__ = ["minmax_to_xywh_torch"]


[docs]def minmax_to_xywh_torch(boxes: torch.Tensor) -> torch.tensor: """ :param boxes: :type boxes: :return: :rtype:""" xmin, ymin, xmax, ymax = boxes.unbind(1) return torch.stack((xmin, ymin, xmax - xmin, ymax - ymin), dim=1)