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