标签： tensorflow keras deep-learning image-segmentation
我有一个用例，需要计算两个不同的损失。一种是正常的前景-背景分割，另一种是主要在边界处细化分割质量。我要包括两个不同的损失函数，每个函数。我知道我们可以将蒙版和输入的RGB图像一起传递给Keras ImageDataGenerator进行训练。现在，问题是可以同时将两个不同的蒙版传递给ImageDataGenerator，以使用相同的模型对两个损失进行计算。