temp change

3f5b71d8 · Kaushik Shivakumar · 94886747 · 3f5b71d8
Commit 3f5b71d8 authored Jul 16, 2020 by Kaushik Shivakumar
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

research/object_detection/dataset_tools/create_ava_tf_record.py ...ch/object_detection/dataset_tools/create_ava_tf_record.py +2 -0

No files found.
--- a/research/object_detection/dataset_tools/create_ava_tf_record.py
+++ b/research/object_detection/dataset_tools/create_ava_tf_record.py
@@ -316,6 +316,8 @@ class Ava(object):
                  feature_list_feature(total_is_annotated) #all ones
          }

+          assert len(total_xmins) == seconds_per_sequence
+
          if len(total_xmins) > 0:
            yield tf.train.SequenceExample(
                context=tf.train.Features(feature=context_feature_dict),