jianlong-yuan / lovaszsoftmax_tf Goto Github PK

您好！yuan！我正在将lovaszsoftmax 应用在我的Unet网络中，但是我遇到了一些问题。在我的网络中，probas、label均为[B,H*W,C]的tensor，我应该如何修改我的网络或者这里的损失函数？我首先尝试了将我的probals reshape为[B,H,W,C],label读入网络改为[B,H,W],但是会遇到错误。希望您能解释下lovaszsoftmax loss中输入的probas、label的含义及格式，以及您的一些建议

There is a “ignore” parameter in your code.
lovasz_loss=lovasz_softmax(tf.nn.softmax(logits),labels,classes='all',per_image=False,ignore=ignore_label)
How do I set this parameter?