Avoid ValueError for TFDS with BBox on the last pixel (blue-oil#1209)

## What this patch does to fix the issue. Fix blue-oil#1208 ## Link to any relevant issues or pull requests.
hadusam · Sep 24, 2020 · ee53474 · ee53474
1 parent f65fac0
commit ee53474
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