search

aleju / imgaug

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

Multiple segmentation maps on one image #633

Closed Angel-Jia closed 4 years ago

Angel-Jia commented 4 years ago

I need bounding box and segmentation at the same time. If two adjacent objects on one image belong to the same category, I cannot separate them from each other after augmentation using SegmentationMapsOnImage. What I need is one segmentation map for one object and multiple segmentaion maps on one image. Is there any solutions?

aleju commented 4 years ago

SegmentationMapsOnImage accepts arrays of shape (H, W, C), which kind of allows you to provide multiple segmentation map arrays via multiple channels. Is that what you want?

Angel-Jia commented 4 years ago

SegmentationMapsOnImage accepts arrays of shape (H, W, C), which kind of allows you to provide multiple segmentation map arrays via multiple channels. Is that what you want?

Thank you very much.