Merge pull request #3744 from liangxiao05/master

Update create_pascal_tf_record.py

Merge pull request #3744 from liangxiao05/master
Update create_pascal_tf_record.py
e3bfb9b8 · pkulzc · GitHub · 2506c9a6 · e658d645 · e3bfb9b8
Unverified Commit e3bfb9b8 authored Apr 09, 2018 by pkulzc Committed by GitHub Apr 09, 2018
Show whitespace changes
Inline Side-by-side

Showing with 17 additions and 15 deletions

research/object_detection/dataset_tools/create_pascal_tf_record.py ...object_detection/dataset_tools/create_pascal_tf_record.py +17 -15

No files found.
--- a/research/object_detection/dataset_tools/create_pascal_tf_record.py
+++ b/research/object_detection/dataset_tools/create_pascal_tf_record.py
@@ -104,6 +104,8 @@ def dict_to_tf_example(data,
  truncated = []
  poses = []
  difficult_obj = []
+    
+  if data.has_key('object'):
    for obj in data['object']:
      difficult = bool(int(obj['difficult']))
      if ignore_difficult_instances and difficult: