2020
10-08
10-08
解决Keras的自定义lambda层去reshape张量时model保存出错问题
前几天忙着参加一个AIChallenger比赛,一直没有更新博客,忙了将近一个月的时间,也没有取得很好的成绩,不过这这段时间内的确学到了很多,就在决赛结束的前一天晚上,准备复现使用一个新的网络UPerNet的时候出现了一个很匪夷所思,莫名其妙的一个问题。谷歌很久都没有解决,最后在一个日语网站上看到了解决方法。事后想想,这个问题在后面搭建网络的时候会很常见,但是网上却没有人提出解决办法,So,Ithinkthat'sverynecess...
继续阅读 >
我就废话不多说了,大家还是直接看代码吧!fromtensorflow.python.keras.modelsimportSequential,Modelfromtensorflow.python.keras.layersimportDense,Flatten,Conv2D,MaxPool2D,Dropout,Conv2DTranspose,Lambda,Input,Reshape,Add,Multiplyfromtensorflow.python.keras.optimizersimportAdamdefdeconv(x):height=x.get_shape()[1].valuewidth=x.get_shape()[2].valuenew_height=height*2n...