fiftyone.utils.tf.TFRecordsWriter

class documentation

class TFRecordsWriter(object): (source)

Class for writing TFRecords to disk.

Example Usage:

with TFRecordsWriter("/path/for/tf.records", num_shards=5) as writer:
    for tf_example in tf_examples:
        writer.write(tf_example)

Parameters
tf_records_path	the path to write the `.tfrecords` file. If sharding is requested `-%%05d-of-%%05d` is appended to the path
num_shards	an optional number of shards to split the records into (using a round robin strategy). If omitted, no sharding is used

Method	`__enter__`	Undocumented
Method	`__exit__`	Undocumented
Method	`__init__`	Undocumented
Method	`write`	Writres the `tf.train.Example` proto to disk.
Instance Variable	`num_shards`	Undocumented
Instance Variable	`tf_records_path`	Undocumented
Instance Variable	`_idx`	Undocumented
Instance Variable	`_num_shards`	Undocumented
Instance Variable	`_writers`	Undocumented
Instance Variable	`_writers_context`	Undocumented