Python Keras-类_重量误差

我正在研究一个脑损伤分割问题，我正试图用以下代码实现一个Unet：

我试图克服的一个问题是类平衡（更多的非损伤体素而不是损伤体素）。我尝试在模型拟合期间使用class_weight，但出现以下错误：

ValueError:3+维目标不支持class_重量

它认为512x512图像是二维的还是262144（512*512）维的。请原谅，如果这是在其他地方解释的话，我对keras是新手。我花了几个小时寻找这个问题，但没有找到令人满意的答案

此外，如果您有关于如何处理此问题的建议（如使用不同的损失函数），请让我知道

---

复制错误的基本代码：

请查看uschmidt83对此github问题的回复。希望有帮助