zizhaozhang/unet-tensorflow-keras

这个代码可以用来分割多类吗?

Kaido0 opened this issue · 5 comments

这个代码可以用来分割多类吗?

可以。需要将conv10改为输出多个channels(类的个数)。然后用softmax + cross_entropy loss

zhcv commented

我多跑分类不行啊,标签怎么制作呢

set opt.num_class to the one you want. The label is a HxW map where each pixel has value [0,...,#of class).

zhcv commented

I have tried , it does not work , did you run multi label segment with this repository?