语义分割损失函数系列(1):交叉熵损失函数这里注意:
最近一直在做一些语义分割相关的项目,找损失函数的时候发现网上这些大佬的写得各有千秋,也没说怎么用,在此记录一下自己在训练过程中使用损失函数的一些心得.本人是使用的Pytorch框架,故这一系列都会基于Pytorch来实现。首先是交叉熵损失函数,语义分割其实是一个逐像素分类的一个分类问题,做过图像分类的应该都比较熟悉交叉熵损失函数。pytorch中自带有写好的交叉熵函数,只需要调用就行:loss_func = nn.CrossEntropyLoss()torch.nn模块中写好的损失函数都是以类的