Keras - class_weight错误

时间:2017-06-25 04:11:02

标签: python deep-learning keras image-segmentation data-science

我正致力于解决脑部病变细分问题,并尝试使用受以下代码启发的代码实现Unet:https://github.com/jocicmarko/ultrasound-nerve-segmentation

我试图克服的问题之一是课堂平衡(更多的非病变体素而不是病变体素)。我试图在模型拟合期间使用class_weight,但我收到以下错误:

ValueError:3维目标不支持class_weight。

是否认为512x512图像是2维或262144(512 * 512)维。请原谅我,如果在其他地方解释我对keras不熟悉的话。我花了几个小时来解决问题,但没有得到满意的答案。

另外,如果您有关于如何处理此问题的建议(例如使用不同的丢失功能),请告诉我。

重现错误here的基本代码:

1 个答案:

答案 0 :(得分:0)

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