Skip to content

GitLab

Unverified Commit e658d645 authored by liangxiao05's avatar liangxiao05 Committed by GitHub
Browse files

Merge pull request #1 from liangxiao05/liangxiao05-patch-1 

Update create_pascal_tf_record.py
parents aeb8cfcb 1cf0a85b
Show whitespace changes
Inline Side-by-side
Showing with 17 additions and 15 deletions
research/object_detection/dataset_tools/create_pascal_tf_record.py
View file @ e658d645
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment