| | |
| | |
| | |
| | |
| | |
| | |
| |
|
| |
|
| | class Bbox: |
| | """ |
| | The class defines Bbox |
| | |
| | Args: |
| | box: The box coordinates |
| | mode: The mode of the box |
| | |
| | Returns: |
| | Bbox: The Bbox object |
| | """ |
| |
|
| | def __init__(self, box, mode="whwh"): |
| | """ |
| | Initializes the Bbox object. |
| | |
| | Args: |
| | box: The box coordinates |
| | mode: The mode of the box |
| | """ |
| | assert len(box) == 4 |
| | assert mode in ["whwh", "xywh"] |
| | self.box = box |
| | self.mode = mode |
| |
|
| | def to_xywh(self): |
| | """ |
| | Converts the box to xywh mode. |
| | |
| | Returns: |
| | Bbox: The Bbox object |
| | """ |
| |
|
| | if self.mode == "whwh": |
| |
|
| | l, t, r, b = self.box |
| |
|
| | center_x = (l + r) / 2 |
| | center_y = (t + b) / 2 |
| | width = r - l |
| | height = b - t |
| | return Bbox([center_x, center_y, width, height], mode="xywh") |
| | else: |
| | return self |
| |
|
| | def to_whwh(self): |
| | """ |
| | Converts the box to whwh mode. |
| | |
| | Returns: |
| | Bbox: The Bbox object |
| | """ |
| |
|
| | if self.mode == "whwh": |
| | return self |
| | else: |
| |
|
| | cx, cy, w, h = self.box |
| | l = cx - w // 2 |
| | t = cy - h // 2 |
| | r = cx + w - (w // 2) |
| | b = cy + h - (h // 2) |
| |
|
| | return Bbox([l, t, r, b], mode="whwh") |
| |
|
| | def area(self): |
| | """ |
| | Calculates the area of the box. |
| | |
| | Returns: |
| | float: The area of the box |
| | """ |
| |
|
| | box = self.to_xywh() |
| | _, __, w, h = box.box |
| |
|
| | return w * h |
| |
|
| | def offset(self, offset_w, offset_h): |
| | """ |
| | Offsets the box by the given width and height. |
| | |
| | Args: |
| | offset_w: The width offset |
| | offset_h: The height offset |
| | |
| | """ |
| |
|
| | assert self.mode == "whwh" |
| |
|
| | self.box[0] += offset_w |
| | self.box[1] += offset_h |
| | self.box[2] += offset_w |
| | self.box[3] += offset_h |
| |
|
| | def get_box(self): |
| | """ |
| | Returns the bounding box as a list of integers. |
| | |
| | Returns: |
| | list: The bounding box values as a list of ints |
| | """ |
| |
|
| | return list(map(int, self.box)) |
| |
|
| | def to_xywh_ratio(self, ori_w, ori_h): |
| | """ |
| | Converts the box to xywh mode and returns the ratio of the box to the original image. |
| | |
| | Args: |
| | ori_w: The original width |
| | ori_h: The original height |
| | |
| | Returns: |
| | tuple: The ratio of the box to the original image |
| | """ |
| |
|
| | cx, cy, w, h = self.to_xywh().get_box() |
| | cx = cx / ori_w |
| | cy = cy / ori_h |
| | w = w / ori_w |
| | h = h / ori_h |
| |
|
| | return cx, cy, w, h |
| |
|
| | def scale_bbox(self, ori_w, ori_h, new_w, new_h): |
| | """ |
| | Scales the box as the image scale. |
| | |
| | Args: |
| | ori_w: The original width |
| | ori_h: The original height |
| | new_w: The new width |
| | new_h: The new height |
| | |
| | Returns: |
| | Bbox: The scaled box |
| | """ |
| |
|
| |
|
| | assert self.mode == "whwh" |
| |
|
| | cx, cy, w, h = self.to_xywh_ratio(ori_w, ori_h) |
| |
|
| | cx = cx * new_w |
| | cy = cy * new_h |
| | w = w * new_w |
| | h = h * new_h |
| |
|
| | l = cx - w // 2 |
| | t = cy - h // 2 |
| | r = cx + w - (w // 2) |
| | b = cy + h - (h // 2) |
| |
|
| | return Bbox([l, t, r, b], mode="whwh") |
| |
|
| | def scale(self, scale, width, height): |
| | """ |
| | Scales the box with the given scale factor. |
| | |
| | Args: |
| | scale: The scale factor |
| | width: The width of the image |
| | height: The height of the image |
| | |
| | Returns: |
| | Bbox: The scaled box |
| | """ |
| |
|
| | new_box = self.to_xywh() |
| | cx, cy, w, h = new_box.get_box() |
| | w = w * scale |
| | h = h * scale |
| |
|
| | l = cx - w // 2 |
| | t = cy - h // 2 |
| | r = cx + w - (w // 2) |
| | b = cy + h - (h // 2) |
| |
|
| | l = int(max(l, 0)) |
| | t = int(max(t, 0)) |
| | r = int(min(r, width)) |
| | b = int(min(b, height)) |
| |
|
| | return Bbox([l, t, r, b], mode="whwh") |
| |
|
| | def __repr__(self): |
| | box = self.to_whwh() |
| | l, t, r, b = box.box |
| |
|
| | return f"BBox(left={l}, top={t}, right={r}, bottom={b})" |
| |
|
