首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Keras中的多类分割

Keras中的多类分割
EN

Stack Overflow用户
提问于 2020-12-05 10:54:17
回答 1查看 255关注 0票数 0

我试图在Keras中实现多类分段:

  • 输入图像为灰度(即1通道)
  • 地面真实图像有3个通道,每个像素是一个长度为3的热矢量。
  • 预测是标准的U-网训练用categorical_crossentropy输出3通道(软件最大编辑)

这个设置有什么问题吗?训练失败有一些奇怪的行为:

  • 在我幸运的情况下,它的行为符合预期(下降)
  • 90 %的时间被困在0.9左右

我的实现可以找到这里

我不认为代码有什么问题:如果我的基本真理是1通道(即到处都是0和某处的1),并且使用binary_crossentropy + sigmoid作为最终的激活,我看不出有什么奇怪的行为。

EN

回答 1

Stack Overflow用户

发布于 2020-12-05 13:23:36

我自己回答问题。解决方案是对每一类进行加权,即使用加权交叉熵损失。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/65156144

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档