相同的网络结构,相同的数据,相同的优化器,相同的损失函数,但是keras和tens之间的值不同

2024-06-10 05:19:50 发布

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

我用keras构建了一个“unet”来进行图像分割,然后通过“fit_generator”来训练网络,在验证数据上得到了0.91的dice_coe,在训练数据上得到了0.92的dice_coe,然而,当用tensorflow来训练相同的“unet”(构建网络的代码不变)时,过度拟合是非常严重的。那么,如何通过tensorflow和keras得到相同的结果呢


Tags: 数据代码图像网络unettensorflowdicegenerator