Skip to content

GitLab

Commit 3f5b71d8 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

temp change

parent 94886747
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
research/object_detection/dataset_tools/create_ava_tf_record.py
View file @ 3f5b71d8
...@@ -316,6 +316,8 @@ class Ava(object): ...@@ -316,6 +316,8 @@ class Ava(object):
feature_list_feature(total_is_annotated) #all ones feature_list_feature(total_is_annotated) #all ones
} }
assert len(total_xmins) == seconds_per_sequence
if len(total_xmins) > 0: if len(total_xmins) > 0:
yield tf.train.SequenceExample( yield tf.train.SequenceExample(
context=tf.train.Features(feature=context_feature_dict), context=tf.train.Features(feature=context_feature_dict),
......
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