augraphy icon indicating copy to clipboard operation
augraphy copied to clipboard

fix OneOf and AugmentationsSequence to process bboxes properly

Open Travvy88 opened this issue 3 months ago • 0 comments

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 filter OneOf or AugmentationSequence because these classes are children of 'Augmentation' class.
  • Make AugmentationSequence return bboxes and other info inside result variable.
  • In case when AugmentationSequence is inside OneOf object I added to OneOf some code to return bboxes and other info accurately.

Travvy88 avatar Oct 29 '24 09:10 Travvy88