mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-26 20:14:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(BoundingBox): Fixing the BoundingBox.area() method to work for all values of CoordOrigin

Browse Source 
Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>
This commit is contained in:
Nikos Livathinos 2024-10-07 15:41:35 +02:00
parent 9eb3afc16c
commit a9b22a8694
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docling/datamodel/base_models.py
View File

@ -110,7 +110,10 @@ class BoundingBox(BaseModel):
return BoundingBox(l=l, t=t, r=r, b=b, coord_origin=origin) return BoundingBox(l=l, t=t, r=r, b=b, coord_origin=origin)
def area(self) -> float: def area(self) -> float:
return (self.r - self.l) * (self.b - self.t) area = (self.r - self.l) * (self.b - self.t)
if self.coord_origin == CoordOrigin.BOTTOMLEFT:
area = -area
return area
def intersection_area_with(self, other: "BoundingBox") -> float: def intersection_area_with(self, other: "BoundingBox") -> float:
# Calculate intersection coordinates # Calculate intersection coordinates