Skip to content

GitLab

Commit 63ec7359 authored by Sara Beery's avatar Sara Beery Committed by TF Object Detection Team
Browse files

Adding float feature to dataset_util 

PiperOrigin-RevId: 351610384
parent 7a8cbff1
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
research/object_detection/utils/dataset_util.py
View file @ 63ec7359
...@@ -38,6 +38,10 @@ def bytes_list_feature(value): ...@@ -38,6 +38,10 @@ def bytes_list_feature(value):
return tf.train.Feature(bytes_list=tf.train.BytesList(value=value)) return tf.train.Feature(bytes_list=tf.train.BytesList(value=value))
def float_feature(value):
return tf.train.Feature(float_list=tf.train.FloatList(value=[value]))
def float_list_feature(value): def float_list_feature(value):
return tf.train.Feature(float_list=tf.train.FloatList(value=value)) return tf.train.Feature(float_list=tf.train.FloatList(value=value))
......
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