File size: 508 Bytes
995526c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | """
Body part annotation data class used by the mask finalization step.
"""
class BodyPart:
"""Represents a detected body part with bounding box, center, and dimensions."""
def __init__(self, name, xmin, ymin, xmax, ymax, x, y, w, h):
self.name = name
# Bounding box
self.xmin = xmin
self.ymin = ymin
self.xmax = xmax
self.ymax = ymax
# Center
self.x = x
self.y = y
# Dimensions
self.w = w
self.h = h
|