update weighted softmax loss

contrib/HumanSeg/nets/seg_modules.py

@@ -34,6 +34,7 @@ def softmax_with_loss(logit,
         loss, probs = fluid.layers.softmax_with_cross_entropy(
             logit, label, ignore_index=ignore_index, return_softmax=True)
     else:
+        label = fluid.layers.squeeze(label, axes=[-1])
         label_one_hot = fluid.one_hot(input=label, depth=num_classes)
         if isinstance(weight, list):
             assert len(

contrib/RemoteSensing/nets/loss.py

@@ -34,6 +34,7 @@ def softmax_with_loss(logit,
         loss, probs = fluid.layers.softmax_with_cross_entropy(
             logit, label, ignore_index=ignore_index, return_softmax=True)
     else:
+        label = fluid.layers.squeeze(label, axes=[-1])
         label_one_hot = fluid.one_hot(input=label, depth=num_classes)
         if isinstance(weight, list):
             assert len(

pdseg/loss.py

@@ -40,6 +40,7 @@ def softmax_with_loss(logit,
             ignore_index=cfg.DATASET.IGNORE_INDEX,
             return_softmax=True)
     else:
+        label = fluid.layers.squeeze(label, axes=[-1])
         label_one_hot = fluid.one_hot(input=label, depth=num_classes)
         if isinstance(weight, list):
             assert len(