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 filterOneOforAugmentationSequencebecause these classes are children of 'Augmentation' class. - Make
AugmentationSequencereturn bboxes and other info inside result variable. - In case when
AugmentationSequenceis insideOneOfobject I added toOneOfsome code to return bboxes and other info accurately.