search

zhixuhao / unet

unet for image segmentation
MIT License
4.57k stars 2k forks source link

multi labeled data #6

Closed ghost closed 7 years ago

ghost commented 7 years ago

Hi, This model is suitable for binary segmentation. What if I want to do multi-label segmentation?

Best,

zhixuhao commented 7 years ago

you should generate image label mask with shape (n,512,512,m), n is the number of images and m is number of classes. and image is 512*512. you can refer to my responsitory. https://github.com/zhixuhao/unet-multi

zhixuhao commented 7 years ago

here is the data https://github.com/preddy5/segnet/tree/master/CamVid

MainakBhattacharya commented 5 years ago

Hi @zhixuhao , so basically instead of learning 1 mask for an input image, the model should learn m different masks corresponding to m classes?