tf2 update again

rc68 · Aug 3, 2020 · b5e444a · b5e444a
1 parent c6ee311
commit b5e444a
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/models/.DS_Store b/models/.DS_Store
diff --git a/models/object_detection/.DS_Store b/models/object_detection/.DS_Store
diff --git a/models/object_detection/code/generate_tfrecord.py b/models/object_detection/code/generate_tfrecord.py
@@ -104,7 +104,7 @@ def create_tf_example(group, path, label_map):
 
 
 def main(_):
-    writer = tf.python_io.TFRecordWriter(FLAGS.output_path)
+    writer = tf.compat.v1.python_io.TFRecordWriter(FLAGS.output_path)
     path = os.path.join(os.getcwd(), FLAGS.img_path)
     examples = pd.read_csv(FLAGS.csv_input)