".github/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "943464ccb85678069b2ad44051f13203c51e67b0"
Unverified Commit e3bfb9b8 authored by pkulzc's avatar pkulzc Committed by GitHub
Browse files

Merge pull request #3744 from liangxiao05/master

Update create_pascal_tf_record.py
parents 2506c9a6 e658d645
...@@ -104,6 +104,8 @@ def dict_to_tf_example(data, ...@@ -104,6 +104,8 @@ def dict_to_tf_example(data,
truncated = [] truncated = []
poses = [] poses = []
difficult_obj = [] difficult_obj = []
if data.has_key('object'):
for obj in data['object']: for obj in data['object']:
difficult = bool(int(obj['difficult'])) difficult = bool(int(obj['difficult']))
if ignore_difficult_instances and 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