Skip to main content

TFRecords

Simple standard + tools for I/O of TFRecords

pip install tensorflow-records

I/O

import tensorflow as tf
import tf.records as tfr

image = tf.ones([1024, 768, 3], dtype=tf.uint8)
label = tf.constant('cat')

spec = tfr.spec(
  image=tfr.Tensor([1024, 768, 3], dtype='int'),
  label='string'
)

serialized = tfr.serialize(spec, image=x, label=y)
# b'\n+\n\x10\n\x05label\x12\x07\n\x05\n\x03cat\n\x17\n\x05image\x1...'

tfr.parse(spec).sample(tf.constant(serialized))
# { 'image': <tf.Tensor: shape=(1024, 768, 3) ...>, 'label': <tf.Tensor: shape=() ...> }


# or you can parse multiple at once
tfr.parse(spec).batch(tf.constant([serialized, serialized]))
# { 'image': <tf.Tensor: shape=(2, 1024, 768, 3) ...>, 'label': <tf.Tensor: shape=(2,) ...> }

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tensorflow_records-0.1.14.tar.gz (5.0 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

tensorflow_records-0.1.14-py3-none-any.whl (6.3 kB view details)

Uploaded Python 3

File details

Details for the file tensorflow_records-0.1.14.tar.gz.

File metadata

  • Download URL: tensorflow_records-0.1.14.tar.gz
  • Upload date:
  • Size: 5.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.0 CPython/3.11.9

File hashes

Hashes for tensorflow_records-0.1.14.tar.gz
Algorithm Hash digest
SHA256 8fc4c42044b77a1a0b007b305a690b7f5e80db9da90aeff94ec6c3c1487e944a
MD5 921a7de094fc0e48cf3bd47f42691ba9
BLAKE2b-256 b8991619964cb4dc7e54849eeaf8922592d13dcce34bcf39e8163b9f4f61586f

See more details on using hashes here.

File details

Details for the file tensorflow_records-0.1.14-py3-none-any.whl.

File metadata

File hashes

Hashes for tensorflow_records-0.1.14-py3-none-any.whl
Algorithm Hash digest
SHA256 d46d391a634a00d3ad6c79e0d51a622b3ea210ead8d0e3eba1ae7e4fa990e0d4
MD5 2a726b53dbd439ed55e592eb3231b547
BLAKE2b-256 10f7476f9da667ffc90df13fa6d51b91b1ee6beea4d357dec7739c818f0a9f78

See more details on using hashes here.

Release history Release notifications | RSS feed

This release

0.1.14 This release

2 files

0.1.13

2 files

0.1.12

2 files

0.1.11

2 files

0.1.9

2 files

0.1.7

2 files

0.1.6

2 files

0.1.5

2 files

0.1.4

2 files

0.1.3

2 files

0.1.2

2 files

0.1.1

2 files

0.1.0

2 files

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page