【tensorflow】AttributeError: module ‘tensorflow._api.v1.compat‘ has no attribute ‘v2‘
2020年2月4日 0次阅读 共391个字 0条评论 0人点赞 QueenDekimZ
tensorflow.slim
python object_detection/builders/model_builder_test.py
出现报错：
将代码
def global_pool(input_tensor, pool_op=tf.compat.v2.nn.avg_pool2d):
改为如下
def global_pool(input_tensor, pool_op=tf.nn.avg_pool2d):
依旧出现报错：
查看源代码后发现：
tf.nn模块中没有avg_pool2d。
查找docs后发现tensorflow.contrib.slim模块中有avg_pool2d
将代码改为如下：
def global_pool(input_tensor, pool_op=tf.contrib.slim.avg_pool2d):
最终运行成功：
