making label

[zengjianyou]

Your implementation is very good, why don't you use the background as a segmentation category? FCN uses one-hot is often to use the background as a category

[usuyama]

Thanks for your comment. The synthetic data example in this repo uses multi-label segmentation - i.e. objects can be overlapped so one pixel can have multiple positive labels. For multi-class segmentation / one-hot setting, I think it's a good idea to have a background category.