Spaces:
Running
Running
File size: 3,440 Bytes
67a9b5d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import numpy as np
def convert_xyxy_to_xywh(boxes, copy=True):
"""Convert [x_min, y_min, x_max, y_max] format to [x_min, y_min, width, height] format.
"""
if copy:
boxes = boxes.copy()
boxes[..., 2:4] -= boxes[..., 0:2]
return boxes
def convert_xywh_to_xyxy(boxes, copy=True):
"""Convert [x_min, y_min, width, height] format to [x_min, y_min, x_max, y_max] format.
"""
if copy:
boxes = boxes.copy()
boxes[..., 2:4] += boxes[..., 0:2]
return boxes
def convert_xywh_to_cxcywh(boxes, copy=True):
"""Convert [x_min, y_min, width, height] format to [cx, cy, width, height] format.
"""
if copy:
boxes = boxes.copy()
boxes[..., 0:2] += boxes[..., 2:4] * 0.5
return boxes
def convert_cxcywh_to_xywh(boxes, copy=True):
"""Convert [cx, cy, width, height] format to [x_min, y_min, width, height] format.
"""
if copy:
boxes = boxes.copy()
boxes[..., 0:2] -= boxes[..., 2:4] * 0.5
return boxes
def convert_xyxy_to_cxcywh(boxes, copy=True):
"""Convert [x_min, y_min, x_max, y_max] format to [cx, cy, width, height] format.
"""
if copy:
boxes = boxes.copy()
boxes[..., 2:4] -= boxes[..., 0:2]
boxes[..., 0:2] += boxes[..., 2:4] * 0.5
return boxes
def convert_cxcywh_to_xyxy(boxes, copy=True):
"""Convert [cx, cy, width, height] format to [x_min, y_min, x_max, y_max] format.
"""
if copy:
boxes = boxes.copy()
boxes[..., 0:2] -= boxes[..., 2:4] * 0.5
boxes[..., 2:4] += boxes[..., 0:2]
return boxes
def convert_boxes_format(boxes, in_fmt, out_fmt, copy=True):
"""Converts boxes from given in_fmt to out_fmt.
Supported in_fmt and out_fmt are:
'xyxy': boxes are represented via corners, x1, y1 being top left and x2, y2 being bottom right.
'xywh' : boxes are represented via corner, width and height, x1, y2 being top left, w, h being width and height.
'cxcywh' : boxes are represented via centre, width and height, cx, cy being center of box, w, h
being width and height.
Args:
boxes: boxes which will be converted.
in_fmt (str): Input format of given boxes. Supported formats are ['xyxy', 'xywh', 'cxcywh'].
out_fmt (str): Output format of given boxes. Supported formats are ['xyxy', 'xywh', 'cxcywh']
Returns:
boxes: Boxes into converted format.
References:
torchvision.ops.box_convert
"""
allowed_fmts = ("xyxy", "xywh", "cxcywh")
if in_fmt not in allowed_fmts or out_fmt not in allowed_fmts:
raise ValueError("Unsupported Bounding Box Conversions for given in_fmt and out_fmt")
if copy:
boxes = boxes.copy()
if in_fmt == out_fmt:
return boxes
if (in_fmt, out_fmt) == ("xyxy", "xywh"):
boxes = convert_xyxy_to_xywh(boxes, copy=False)
elif (in_fmt, out_fmt) == ("xywh", "xyxy"):
boxes = convert_xywh_to_xyxy(boxes, copy=False)
elif (in_fmt, out_fmt) == ("xywh", "cxcywh"):
boxes = convert_xywh_to_cxcywh(boxes, copy=False)
elif (in_fmt, out_fmt) == ("cxcywh", "xywh"):
boxes = convert_cxcywh_to_xywh(boxes, copy=False)
elif (in_fmt, out_fmt) == ("xyxy", "cxcywh"):
boxes = convert_xyxy_to_cxcywh(boxes, copy=False)
elif (in_fmt, out_fmt) == ("cxcywh", "xyxy"):
boxes = convert_cxcywh_to_xyxy(boxes, copy=False)
return boxes
|