diff --git a/blueoil/utils/tfds_builders/object_detection.py b/blueoil/utils/tfds_builders/object_detection.py index 280c85354..17f7e30d5 100644 --- 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