Quantize bboxes

Author: feynmanliang

labelbox/exporters/pascal_voc_writer/__init__.py

@@ -38,8 +38,12 @@ def add_object(self, name, xy_coords):
         if len(xy_coords) == 8:
             x_points = sorted(xy_coords[::2])
             y_points = sorted(xy_coords[1::2])
-            if (x_points[0] == x_points[1] and x_points[2] == x_points[3]
-                    and y_points[0] == y_points[1] and y_points[2] == y_points[3]):
+
+            # quantize up to 1 px off to bounding boxes
+            if (abs(x_points[0] - x_points[1]) < 1 and abs(x_points[2] - x_points[3]) < 1
+                    and abs(y_points[0] == y_points[1]) < 1 and abs(y_points[2] == y_points[3]) < 1:
+                x_points[0] = x_points[1]
+                x_points[2] = x_points[3]
                 label_type = 'bndbox'
             else:
                 label_type = 'polygon'