AttributeError: módulo 'tensorflow' não tem atributo 'get_default_graph' em tensorflow

Aug 18 2020

Estou recebendo o seguinte erro no meu código.

import mtcnn
# print version
print(mtcnn.__version__)

# demonstrate face detection on 5 Celebrity Faces Dataset
from os import listdir
from PIL import Image
from numpy import asarray
from matplotlib import pyplot
from mtcnn.mtcnn import MTCNN
print("MTCNN: {}".format(mtcnn.__version__))
import tensorflow as tf
from tensorflow import keras

# extract a single face from a given photograph
def extract_face(filename, required_size=(160, 160)):
    # load image from file
    image = Image.open(filename)
    # convert to RGB, if needed
    image = image.convert('RGB')
    # convert to array
    pixels = asarray(image)
    # create the detector, using default weights
    detector = MTCNN()
    # detect faces in the image
    results = detector.detect_faces(pixels)
    # extract the bounding box from the first face
    x1, y1, width, height = results[0]['box']
    # bug fix
    x1, y1 = abs(x1), abs(y1)
    x2, y2 = x1 + width, y1 + height
    # extract the face
    face = pixels[y1:y2, x1:x2]
    # resize pixels to the model size
    image = Image.fromarray(face)
    image = image.resize(required_size)
    face_array = asarray(image)
    return face_array

# specify folder to plot
#folder = '5-celebrity-faces-dataset/train/ben_afflek/'
folder = '5-celebrity-faces-dataset/train/ben_afflek'
i = 1
# enumerate files
for filename in listdir(folder):
    # path
    path = folder + '/' + filename
    # get face
    face = extract_face(path)
    print(i, face.shape)
    # plot
    pyplot.subplot(2, 7, i)
    pyplot.axis('off')
    pyplot.imshow(face)
    i += 1
pyplot.show()

Erro:

anaconda3\envs\py3\lib\site-packages\keras\backend\tensorflow_backend.py", linha 68, em gráfico get_uid = tf.get_default_graph()

AttributeError: módulo 'tensorflow' não tem atributo 'get_default_graph'

Eu tentei várias importações diferentes, mas nada funciona. Parece que esse erro é normal, mas não estou encontrando nada que resolva meu problema.

Respostas

1 TimbusCalin Aug 18 2020 at 13:32

Na resposta aqui importar keras do tensorflow, como você fez, resolve o problema.

Mas o problema no seu caso é que o MTCNN funciona no Keras puro ao invés do TensorFlow, então o fato de você carregar no seu keras "main.py" do tensorflow não tem nenhum efeito. Você precisa fazer o downgrade da versão do tensorflow ou modificar todas as importações no MTCNN, o que não é garantido para funcionar, infelizmente.