search

mdbloice / Augmentor

Image augmentation library in Python for machine learning.
https://augmentor.readthedocs.io/en/stable
MIT License
5.07k stars 866 forks source link

How to make this work with multiple masks? #87

Open CMCDragonkai opened 6 years ago

CMCDragonkai commented 6 years ago

For any given image, there may be multiple masks (for used in instance segmentation). How does the ground_truth function work with this?

mdbloice commented 6 years ago

Hi @CMCDragonkai. Right now, Augmentor works with one mask. However, all the operations are actually capable of taking an arbitrarily long list of images to augment identically (this was part of a rewrite for version 0.2.0, as previously all operations worked on single images only), so this is a feature I plan to add very soon. I just need to think of a nice way to specify several masks.

VikramRadhakrishnan commented 6 years ago

HI @mdbloice . Has this implementation been added to Augmentor yet? Is it capable of taking a ground truth image plus a list of masks?

zhaoxin111 commented 6 years ago

hi@mdbloice , does the multi masks augmentation finished yet?

FiReTiTi commented 6 years ago

Hi, I also need it. Has it been implemented? If no, would it be difficult if I try to do it myself?

mdbloice commented 6 years ago

Hi @CMCDragonkai, @VikramRadhakrishnan, @zhaoxin111, @FiReTiTi, the multi-mask feature is very close to being completed, I have been working on it the last couple of weeks, and I should have a version up in the next few days.

FiReTiTi commented 6 years ago

Thanks a lot. Please, let me know when you have something, I'll gladly test it.

mdbloice commented 6 years ago

I'll drop a message here @FiReTiTi once I have pushed the new version on to PyPI!

blaxe05 commented 5 years ago

hii @mdbloice great toolbox!!! by the way, does it support multi-mask feature now, ex: let say 384x384x4 where 4 is the number of different masks?