
keras
Lwang2018
这个作者很懒,什么都没留下…
展开
-
[Keras] 模型可视化 plot_model
Keras中提供了一个神经网络可视化的函数plot_model,并可以将可视化结果保存在本地:from keras.utils.vis_utils import plot_modelmodel = unet()plot_model(model, to_file='model-unet.png')我这里可视化了一个U-net模型...原创 2019-01-28 15:06:55 · 10407 阅读 · 3 评论 -
医学图像分割中常用的loss函数
loss函数的选择是个很大的命题,最近在做3D医学图像分割,我这里只是记录一下我用过的loss函数。Binary Cross Entropy Loss交叉熵损失比较常用,代码实现:from keras import backend as Kdef binary_crossentropy(y_true, y_pred): return K.mean(K.binary_cross...原创 2019-08-11 16:59:16 · 3681 阅读 · 0 评论