Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "fbfa50381b092d30a057b5cc8908bdb2fc1ce6c4"
Commit 35c1908f authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix earlier bug, need to test still

parent 2419ad3b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
research/object_detection/dataset_tools/create_ava_tf_record.py
View file @ 35c1908f
...@@ -241,6 +241,7 @@ class Ava(object): ...@@ -241,6 +241,7 @@ class Ava(object):
total_images = [] total_images = []
total_source_ids = [] total_source_ids = []
total_confidences = [] total_confidences = []
total_is_annotated = []
windowed_timestamp = start_time windowed_timestamp = start_time
while windowed_timestamp < end_time: while windowed_timestamp < end_time:
skipped_frame_count = 0; skipped_frame_count = 0;
...@@ -256,6 +257,7 @@ class Ava(object): ...@@ -256,6 +257,7 @@ class Ava(object):
source_id = str(GLOBAL_SOURCE_ID) + "_" + media_id source_id = str(GLOBAL_SOURCE_ID) + "_" + media_id
total_source_ids.append(dataset_util.bytes_feature( total_source_ids.append(dataset_util.bytes_feature(
source_id.encode("utf8"))) source_id.encode("utf8")))
total_is_annotated.append(dataset_util.int64_feature(1))
GLOBAL_SOURCE_ID += 1 GLOBAL_SOURCE_ID += 1
if (media_id, windowed_timestamp) in frame_excluded: if (media_id, windowed_timestamp) in frame_excluded:
end_time += 1 end_time += 1
...@@ -326,7 +328,7 @@ class Ava(object): ...@@ -326,7 +328,7 @@ class Ava(object):
'region/label/confidence': 'region/label/confidence':
feature_list_feature(total_confidences), #all ones feature_list_feature(total_confidences), #all ones
'region/is_annotated': 'region/is_annotated':
feature_list_feature(total_confidences) #all ones feature_list_feature(total_is_annotated) #all ones
} }
if len(total_xmins) > 0: if len(total_xmins) > 0:
......
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