问题描述:
从网上找了一个大佬的图像处理项目,在运行时出现了几个报错都是module ‘tensorflow’ has no attribute,问题产生的原因是当前Python使用的Tensorflow库为2.0最新版本,而源代码使用的是1.0版本,如何在不降低版本的情况下运行代码呢?
解决方法:
找到报错的地方,在报错的attribute前面加上compat.v1. 代码示例: 源代码:
self.keep_prob = tf.placeholder(tf.float32) # dropout值
改动后:
self.keep_prob = tf.compat.v1.placeholder(tf.float32) # dropout值
再次运行就不会出错了,该方法适用于大多数情况,特殊情况以下给出。
contrib特殊处理
当报错为AttributeError: module ‘tensorflow‘ has no attribute ‘contrib‘ 找到报错的代码,一般为:
initializer = tf.contrib.layers.xavier_initializer()
此时不能用上面的方法修改代码,因为Tensorflow2.0版本把contrib库取消了,因此我们使用tf.initializers.GlorotUniform() 进行初始化,代码改为:
initializer=tf.initializers.GlorotUniform())
成功运行啦!
参考文章
发表评论