2017-06-25 90 views
0

我工作的一個腦損傷分割的問題,我試圖實現與代碼UNET啓發:https://github.com/jocicmarko/ultrasound-nerve-segmentationKeras - class_weight錯誤

一個我試圖克服的問題是職業平衡(更多的非病變體素而不是病變體素)。我試圖在模型適合期間使用class_weight,但出現以下錯誤:

ValueError:class_weight不支持3 +維度目標。

它認爲512x512圖像是2維還是262144(512 * 512)尺寸。請原諒我,如果這是在其他地方解釋我是新來的keras。我花了幾個小時尋找問題,但沒有得到滿意的答案。另外,如果您有關於如何處理此問題的建議(如使用不同的損失函數),請讓我知道。

Basic代碼重現錯誤here

回答