search

wuyang0329 / unet

this is a simple demo for image segmentation.----unet网络进行语义分割的demo，用的数据集是KITTI
97 stars 38 forks source link

Multi-output using U-Net #24

Open tamaraalshekhli opened 4 years ago

tamaraalshekhli commented 4 years ago

Hi there, my input is one image and the out put is 2 masks one for road and the other one for centerline segmentation , for the (trainGenerator) I defined 3 generators, image_generator, mask_generator1 and mask_generator2, now I dont know how to train my Unet using these generators, is there anyone tried multi-labels? any help or advice will be appreciated