Keras等级

2024-05-17 03:19:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在研究一个脑损伤分割问题,我正试图实现一个Unet,代码灵感来自:https://github.com/jocicmarko/ultrasound-nerve-segmentation

我试图克服的一个问题是类平衡(更多的非损伤体素而不是损伤体素)。我试图在模型拟合期间使用类权重,但出现以下错误:

ValueError:三维目标不支持类的权重。在

它认为512x512图像是二维的还是262144(512*512)维的。原谅我,如果这是在其他地方解释的我是新来的凯拉斯。我花了几个小时寻找这个问题,但没有找到一个令人满意的答案。在

另外,如果你有关于如何处理这个问题的建议(比如使用不同的损失函数),请告诉我。在

重现错误here的基本代码:


Tags: 代码https模型githubcomunet错误权重