Maxim Ivanchenko
Maxim Ivanchenko

Reputation: 33

How to fix module 'tensorflow' has no attribute 'get_default_session'

Help solve that problem please I use ImageAI to detect cars , but it didn't works , i'm newbee in this , sorry

from imageai.Detection import ObjectDetection
import os


execution_path = os.getcwd()

detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()

detections, objects_path = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image3.jpg"), output_image_path=os.path.join(execution_path , "image3new.jpg"), minimum_percentage_probability=30,  extract_detected_objects=True)


for eachObject, eachObjectPath in zip(detections, objects_path):
    print(eachObject["name"] , " : " , eachObject["percentage_probability"], " : ", eachObject["box_points"] )
    print("Object's image saved in " + eachObjectPath)
    print("--------------------------------")

Using TensorFlow backend.

Traceback (most recent call last):
File "/home/maxim/PycharmProjects/ImageAI/venv/image.py", line 7, in <module>
    detector = ObjectDetection()
File "/home/maxim/.virtualenvs/ImageAI/lib/python3.5/site-packages/imageai/Detection/__init__.py", line 88, in __init__
    self.sess = K.get_session()
  File "/home/maxim/.virtualenvs/ImageAI/lib/python3.5/site-packages/keras/backend/tensorflow_backend.py", line 174, in get_session
    default_session = tf.get_default_session()
AttributeError: module 'tensorflow' has no attribute 'get_default_session'

Process finished with exit code 1

Upvotes: 3

Views: 3042

Answers (1)

Max Kleiner
Max Kleiner

Reputation: 1612

https://github.com/OlafenwaMoses/ImageAI/issues/563

You're right it (ImageAI) works with Collecting keras==2.1.6, checked on colob, Thanks

Upvotes: 2

Related Questions