使用Deepbrain库时出现错误消息“模块'tensorflow'没有'Session'属性”

Question

我正在尝试使用库Deepbrain从整个磁共振成像扫描中提取大脑，我正在使用代码

def Reduce_Brain(img):
     img_data = img.get_fdata()
     prob = ext.run(img)
     print(prob)

img = nib.load('ADNI_002_S_0295.nii')
Reduce_Brain(img)

然而，当我尝试此操作时，我发现错误模块“tensorflow”没有属性“Session”，我发现这是一个错误，与错误版本的tensorflow有关，因此我随后更改了库代码，如另一个问题中所述（见下文）。但这会产生更多错误，例如模块“tensorflow”没有属性“gfile”

Tensorflow 2.0 - AttributeError: module 'tensorflow' has no attribute 'Session'

https://github.com/iitzco/deepbrain

Answer 1

在tf2.x中，应该使用tf.compat.v1.Session()而不是tf.Session()。 有关更多信息，请查看Migrate_tf2 guide

要在TF2.0中获得类似TF1.x的行为，请添加以下代码

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()