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)])