我试图建立一个模型,采用灰度图像,并生成另一个图像作为输出的残差CNN使用keras。这里的关键思想是,非白色像素的数量远远小于白色像素的数量。因此,每个非白色像素的误差在损失函数中的权重应该大于白色像素的权重。像素越暗,误差应该越大。这就是我如何我现在已经做了。这里,total_pred_score表示所有正确猜测的像素,它们的权重由某个int系数定义,total_true_score表示训练集中所有图像和像素的总期望得分。但我不确定它是否正确。你能帮帮我吗?
def custom_loss(y_true, y_pred):
y_true1, y_pred1 = (255 - y_true) / 255, (255 - y_pred) / 255
dif = (1 - K.abs(y_true1 - y_pred1))
weight = (coeff - 1) * y_true1 + 1
total_true_score = K.sum(weight, axis = [0,1,2,3])
total_pred_score = K.sum(multiply([dif,weight]), axis = [0,1,2,3])
return K.abs(total_true_score - total_pred_score) / total_true_score
1条答案
按热度按时间xsuvu9jc1#
您尝试构建的损失函数应该具有两个属性:
1.每个非白色像素中的误差应比白色像素中的误差具有更大的权重
1.像素越暗,误差越大
因此,为了简化计算,如果将暗像素归为白色像素,则希望误差更大,但对于将白像素归为暗像素,但仍希望它们对损失函数有贡献的情况,不必太担心。此外,您还希望确保损失与像素的强度成正比(像素越暗,误差越大)。
我已经提出了下面的损失函数,它符合这两个性质:
因此,基本上,我们在进行归一化并乘以
dif
(对应于较暗像素)-temp1
中仅与正值成比例的权重之后,找到真实值与预测值之间的差异。然后,我们仅将1
的权重添加到对应于负值(对应于白色)的像素。样本输出
这里,如果你仔细观察
2
和8
位的像素,像素8
对应的情况是,我们预测暗像素为白色像素,这在我们的情况下非常糟糕,因此损失值非常高2.249785
,同样,像素2
对应的情况是,我们预测白色像素为暗像素,这对我们来说还不错,因此损失值为0.98039216
,其不像前一情况那样高。这对应于属性1
。类似地,如果您查看像素
1
和3
,两种情况下的误差幅度相同(相差50
),但像素1
比像素3
暗得多,因此我们希望像素1
上的误差高于像素3
,这就是我们在误差向量中观察到的。在真实值和预测值匹配的情况下,我们有
0
误差。