Keras 模型多输出 loss weight metrics 设置
本文最后更新于:2022年7月4日 上午
keras支持模型多输入多输出,本文记录多输出时loss、loss weight和metrics的设置方式。
模型输出
假设模型具有多个输出
- classify: 二维数组,分类softmax输出,需要配置交叉熵损失
- segmentation:与输入同尺寸map,sigmoid输出,需要配置二分类损失
- others:自定义其他输出,需要自定义损失
具体配置
model
- 变量均为模型中网络层
1 |
|
loss
1 |
|
loss weight
1 |
|
metrics
1 |
|
编译
1 |
|
文章链接:
https://www.zywvvd.com/notes/study/deep-learning/keras/keras-multi-metrics/keras-multi-metrics/
Keras 模型多输出 loss weight metrics 设置
https://www.zywvvd.com/notes/study/deep-learning/keras/keras-multi-metrics/keras-multi-metrics/