问题描述:

从网上找了一个大佬的图像处理项目,在运行时出现了几个报错都是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())

成功运行啦!

参考文章

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: