Fixed the Segmentation fault during generating tfrecord for RNet and …

…ONet.
JellyDream · Jul 23, 2018 · 968f948 · 968f948
1 parent b8631ba
commit 968f948
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/prepare_data/gen_ONet_tfrecords.py b/prepare_data/gen_ONet_tfrecords.py
@@ -63,7 +63,6 @@ def run(dataset_dir, net, output_dir, name='MTCNN', shuffling=False):
             sys.stdout.flush()
             filename = image_example['filename']
             _add_to_tfrecord(filename, image_example, tfrecord_writer)
-    tfrecord_writer.close()
     # Finally, write the labels file:
     # labels_to_class_names = dict(zip(range(len(_CLASS_NAMES)), _CLASS_NAMES))
     # dataset_utils.write_label_file(labels_to_class_names, dataset_dir)

diff --git a/prepare_data/gen_RNet_tfrecords.py b/prepare_data/gen_RNet_tfrecords.py
@@ -63,7 +63,6 @@ def run(dataset_dir, net, output_dir, name='MTCNN', shuffling=False):
             sys.stdout.flush()
             filename = image_example['filename']
             _add_to_tfrecord(filename, image_example, tfrecord_writer)
-    tfrecord_writer.close()
     # Finally, write the labels file:
     # labels_to_class_names = dict(zip(range(len(_CLASS_NAMES)), _CLASS_NAMES))
     # dataset_utils.write_label_file(labels_to_class_names, dataset_dir)