Keras的自定义lambda层去reshape tensor张量时model保存出错的解决办法
背景
分割网络在进行上采样的时候我用的是双线性插值上采样的,而Keras里面并没有实现双线性插值的函数,所以要自己调用tensorflow里面的tf.image.resize_bilinear()函数来进行resize,如果直接用tf.image.resize_bilinear()函数对Keras张量进行resize的话,会报出...
转载
2020-01-12 12:02:11 ·
662 阅读 ·
1 评论