augraphy
augraphy copied to clipboard
fix OneOf and AugmentationsSequence to process bboxes properly
As mentioned in #445, bounding boxes, keypoints and mask is not processed properly when OneOf
or AugmentationSequence
is used. The following issues is fixed:
- Fix
augraphy/base/augmentationpipeline.py:753
. The line do not filterOneOf
orAugmentationSequence
because these classes are children of 'Augmentation' class. - Make
AugmentationSequence
return bboxes and other info inside result variable. - In case when
AugmentationSequence
is insideOneOf
object I added toOneOf
some code to return bboxes and other info accurately.