avoid ValueError for TFDS with BBox on the last pixel

blue-oil · Sep 24, 2020 · f9f8a32 · f9f8a32
1 parent ba646b9
commit f9f8a32
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/blueoil/utils/tfds_builders/object_detection.py b/blueoil/utils/tfds_builders/object_detection.py
@@ -77,8 +77,8 @@ def _generate_examples(self, dataset):
                     "bbox": tfds.features.BBox(
                         ymin / height,
                         xmin / width,
-                        (ymin + h) / height,
-                        (xmin + w) / width,
+                        min((ymin + h) / height, 1.0),
+                        min((xmin + w) / width, 1.0),
                     )
                 }
                 for xmin, ymin, w, h, label in annotations