Od_Graph_Def = Tf. Graphdef() Attributeerror: Module 'Tensorflow' Has No Attribute 'Graphdef'
I Have a Mac and I Am Using Tensorflow 2.0, Python 3.7. I Am Following the Tutorial for Creating an Object Detection Model for Real-Time Application. but I Am...
I have a mac and I am using tensorflow 2.0, python 3.7. I am following the tutorial for creating an object detection model for real-time application. but i am getting the following error:
"Downloads/models/research/object_detection/object_detection_tutorial.py", line 43, in od_graph_def = tf od_graph_def = tf.GraphDef()
AttributeError: module 'tensorflow' has no attribute 'GraphDef'
below is the tutorial link:
I checked the environment and I already have tensorflow environment in anaconda
import tensorflow as tf
import zipfile
from collections import defaultdict
from io import StringIO
from matplotlib import pyplot as plt
from PIL import Image
sys.path.append("..")
from object_detection.utils import ops as utils_ops
from utils import label_map_util
from utils import visualization_utils as vis_util
MODEL_NAME = 'ssd_mobilenet_v1_coco_2017_11_17'
MODEL_FILE = MODEL_NAME + '.tar.gz'
DOWNLOAD_BASE = '
PATH_TO_CKPT = MODEL_NAME + '/frozen_inference_graph.pb'
PATH_TO_LABELS = os.path.join('data', 'mscoco_label_map.pbtxt')
NUM_CLASSES = 90
opener = urllib.request.URLopener()
opener.retrieve(DOWNLOAD_BASE + MODEL_FILE, MODEL_FILE)
tar_file = tarfile.open(MODEL_FILE)
for file in tar_file.getmembers():
file_name = os.path.basename(file.name)
if 'frozen_inference_graph.pb' in file_name:
tar_file.extract(file, os.getcwd())
detection_graph = tf.Graph()
with detection_graph.as_default():
od_graph_def = tf.GraphDef()
with tf.gfile.GFile(PATH_TO_CKPT, 'rb') as fid:
serialized_graph = fid.read()
od_graph_def.ParseFromString(serialized_graph)
tf.import_graph_def(od_graph_def, name='')
3 Answers
Yeah, the syntax has changed in T2.0. Here's the correct piece:
tf.compat.v1.GraphDef() # -> instead of tf.GraphDef()
tf.compat.v2.io.gfile.GFile() # -> instead of tf.gfile.GFile()
I had similar issues, when upgraded to Python 3.7 and Tensorflow 1.2.0 to Tensorflow 2.0.0
If you don't want to touch your code, just add these 2 lines in the main.py file w/ Tensorflow code:
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
And that's it!!
NOW Everything should runs seamlessly :)
But if you write new code, indeed (as was mentioned above) change these calls:
with tf.gfile.GFile(path, 'r') as fid:
To:
with tf.io.gfile.GFile(path, 'r') as fid:
from object_detection.utils import ops as utils_ops
utils_ops.tf = tf.compat.v1
tf.gfile = tf.io.gfile
Adding these lines might fix your problem
Maya is a hardware enthusiast who tests and reviews smart home devices, smartphones, wearables, and audio gear. She focuses on practical consumer value and build quality.
