Skip to content

GitLab

Commit 739a7f32 authored by Vivek Rathod's avatar Vivek Rathod Committed by TF Object Detection Team
Browse files

Set temporal_embedding to None when `data_captured` isn't available. 

PiperOrigin-RevId: 324848229
parent 2db4cc8f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/dataset_tools/context_rcnn/generate_embedding_data.py
View file @ 739a7f32
...@@ -184,7 +184,7 @@ class GenerateEmbeddingDataFn(beam.DoFn): ...@@ -184,7 +184,7 @@ class GenerateEmbeddingDataFn(beam.DoFn):
[unix_time]) [unix_time])
temporal_embedding = embed_date_captured(date_captured) temporal_embedding = embed_date_captured(date_captured)
except Exception: # pylint: disable=broad-except except Exception: # pylint: disable=broad-except
pass temporal_embedding = None
detections = self._detect_fn.signatures['serving_default']( detections = self._detect_fn.signatures['serving_default'](
(tf.expand_dims(tf.convert_to_tensor(tfexample), 0))) (tf.expand_dims(tf.convert_to_tensor(tfexample), 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