Use for Semantic Segmentation

[Superzlw]

Use for Semantic Segmentation

[Superzlw]

Hi @mdbloice , Thanks for the reply. My test code is as follow:

import Augmentor

p = Augmentor.Pipeline(img_path)   # img_path: folder path of the images
p.ground_truth(label_path)         # label_path: folder path of the labels

p.random_erasing(probability=0.8, rectangle_area=0.5)
p.sample(10)

Augmentor works very well for flipping, but for some more complex operations like distortion, erasing, and borders there will be some undesired masks

[NUS-Tim]

Hi, @mdbloice, it seems I also face a similar problem described by @Superzlw .

[mdbloice]

Hi @Superzlw, hi @NUS-Tim

I updated the code in version 0.2.11 of Augmentor just now. Try upgrading with pip install --upgrade Augmentor and let me know if this works for you now... 🤔

[mdbloice]

Hi @Superzlw I will have to take a look at this more closely and try to find a fix... so you are passing images and their masks (as pairs) in to the pipeline, and augmenting them as normal?

If you could show some code that you are using to do this, that would help! If I can see your code I am sure we can fix it.