Keras - class_weight 错误

Question

我正在研究脑损伤分割问题，我正在尝试使用受以下启发的代码实现 Unet：https://github.com/jocicmarko/ultrasound-nerve-segmentation

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

ValueError: class_weight 不支持 3+ 维目标。

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

另外，如果您对如何处理此问题有任何建议（例如使用不同的损失函数），请告诉我。

重现错误here的基本代码：

Answer 1

查看 uschmidt83 对此 github 问题的回复。 https://github.com/fchollet/keras/issues/3653希望有帮助。