search

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.31k stars 1.43k forks source link

如何实现多种分类 #387

Open hhhzr opened 1 week ago

hhhzr commented 1 week ago

修改了out_ch sigmoid改为softmax 损失函数改为nn.CrossEntropyLoss() 仍然无法成功，请求帮助