实现加权交叉熵进行图像分割

Question

我正在尝试对大约680 x 440 RGB图像实施图像分割，以将像素分为4类。我正在使用tensorflow和keras API设计/构建我的模型。问题是我的背景类约为图像的70％，因此使用tensorflow.keras中的标准sparse_crossentropy损失函数时，我的分割结果很差。我想知道如何使用sparse_crossentropy实现加权损失函数，以便可以更好地训练我的模型。