fix: fix bbox conversion to Pascal VOC to use 1-based pixel index

fathomnet · Aug 1, 2024 · 440afb0 · 440afb0
1 parent 67613f9
commit 440afb0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/fathomnet/dto.py b/fathomnet/dto.py
@@ -82,10 +82,10 @@ def to_pascal_voc(
                 E.difficult("0"),
                 E.occluded(str(int(box.occluded) if box.occluded is not None else 0)),
                 E.bndbox(
-                    E.xmin(str(box.x)),
-                    E.xmax(str(box.x + box.width)),
-                    E.ymin(str(box.y)),
-                    E.ymax(str(box.y + box.height)),
+                    E.xmin(str(1 + box.x)),
+                    E.xmax(str(1 + box.x + box.width)),
+                    E.ymin(str(1 + box.y)),
+                    E.ymax(str(1 + box.y + box.height)),
                 ),
             )
             for box in boxes