Skip to content

Commit 1974c9e

Browse files
committed
Fix gt bbox shift in boxes_from_points tasks
1 parent eadc063 commit 1974c9e

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

packages/examples/cvat/exchange-oracle/src/handlers/job_creation.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1465,6 +1465,10 @@ def _prepare_gt_roi_dataset(self):
14651465
categories=self._gt_dataset.categories(), media_type=dm.Image
14661466
)
14671467

1468+
roi_info_by_point_id: dict[int, skeletons_from_boxes_task.RoiInfo] = {
1469+
roi_info.point_id: roi_info for roi_info in self._rois
1470+
}
1471+
14681472
for sample in self._gt_dataset:
14691473
for gt_bbox in sample.annotations:
14701474
assert isinstance(gt_bbox, dm.Bbox)
@@ -1474,10 +1478,15 @@ def _prepare_gt_roi_dataset(self):
14741478
self.escrow_address, self.chain_id, self._roi_filenames[point_id]
14751479
)
14761480

1481+
# update gt bbox coordinates to match RoI shift
1482+
roi_info = roi_info_by_point_id[point_id]
1483+
new_x = gt_bbox.points[0] - roi_info.roi_x
1484+
new_y = gt_bbox.points[1] - roi_info.roi_y
1485+
14771486
self._gt_roi_dataset.put(
14781487
sample.wrap(
14791488
id=os.path.splitext(gt_roi_filename)[0],
1480-
annotations=[gt_bbox],
1489+
annotations=[gt_bbox.wrap(x=new_x, y=new_y)],
14811490
media=dm.Image(path=gt_roi_filename, size=sample.media_as(dm.Image).size),
14821491
attributes=filter_dict(sample.attributes, exclude_keys=["id"]),
14831492
)

0 commit comments

Comments
 (0)