最近在研究keras的cycleGan,將keras轉成tf.keras的版本時發生一個問題:
No module named 'keras_contrib'
看來似乎keras_contrib不存在於tensorflow2.0
取而代之的是tensorflow_addons
下pip指令:pip install tensorflow-addons
然後回到我上面的程式,將
from keras_contrib.layers.normalization.instancenormalization import InstanceNormalization
改成
from tensorflow_addons.layers import InstanceNormalization
就可以正常使用了
沒有留言:
張貼留言
有興趣或有疑問的歡迎提問與交流喔!!!