xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
8.66k stars 1.5k forks source link

u2net for semantic segmentation #382

Open fhfeishi opened 7 months ago

fhfeishi commented 7 months ago

如果说 model = unet_full(out_ch=3), 2类目标+背景 我应该如何修改, cross-entropy loss一直是0