search

aleju / imgaug

Image augmentation for machine learning experiments.
http://imgaug.readthedocs.io
MIT License
14.43k stars 2.44k forks source link

Fix bounding_box assignment on BoundingBoxesOnImage #724

Open semaraugusto opened 4 years ago

semaraugusto commented 4 years ago

While performing augmentations on bounding boxes one might have to use the classes ia.augmentables.bbs.BoundingBox and/or ia.augmentables.bbs.BoundingBoxesOnImage. However, the assignment of the latter does not check if the variable bounding_boxes is of the correct type (a list containing ia.augmentables.bbs.BoundingBox). This change standardizes the object BoundingBoxesOnImage.bounding_boxes to be always a list of BoundingBox objects