我不是100%熟悉TF。但是,您是否考虑过使用weights
损失参数? 看tf.loses.sparse_softmax_cross_entropy
它有一个参数weights
:损失系数。这必须是标量或与标签具有相同的等级
您可以将weight
“无效”像素设置为零,从而使损耗忽略它们。
您也可以从中删除减少量tf.nn.sparse_softmax_cross_entropy_with_logits
并用于tf.losses.compute_weighted_loss
执行加权。