From 24e8c32554879ff9457510cd22c5380b279d9c9f Mon Sep 17 00:00:00 2001
From: Daniel Kusuma <63502967+ksmdnl@users.noreply.github.com>
Date: Sat, 2 Nov 2024 10:22:45 +0100
Subject: [PATCH] bugfix area given Pascal bbox

---
 src/table_datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/table_datasets.py b/src/table_datasets.py
index 1fbe017e7..176a7daa8 100644
--- a/src/table_datasets.py
+++ b/src/table_datasets.py
@@ -646,7 +646,7 @@ def __getitem__(self, idx):
         target["boxes"] = bboxes
         target["labels"] = labels
         target["image_id"] = torch.as_tensor([idx])
-        target["area"] = bboxes[:, 2] * bboxes[:, 3] # COCO area
+        target["area"] = (bboxes[:, 2] - bboxes[:, 0]) * (bboxes[:, 3] - bboxes[:, 1]) # COCO area
         target["iscrowd"] = torch.zeros((num_objs,), dtype=torch.int64)
         target["orig_size"] = torch.as_tensor([int(h), int(w)])
         target["size"] = torch.as_tensor([int(h), int(w)])