search

albumentations-team / albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
https://albumentations.ai
MIT License
13.7k stars 1.61k forks source link

semantic segmentation #766

Open Ciaran1981 opened 3 years ago

Ciaran1981 commented 3 years ago

📚 Documentation

Hello,

This is a question/request rather than an issue. How does one alter the binary segmentation example workflow to do a multi-class semantic segmentation?

I have attempted this for a long time to no avail and it'd be great to see how albumentations can be used for a multi class semantic segmentation.

https://albumentations.ai/docs/examples/pytorch_semantic_segmentation/

adbcode commented 8 months ago

@Ciaran1981 were you able to figure something that worked in your case?

Dipet commented 8 months ago

Just specify a mask for each class and use masks instead of mask