Python parallel_record_writer примеры использования

Пример #1

def _convert_data(split):
  def _create_example(item):
    image, label = item
    example = tf.train.Example(features=tf.train.Features(
      feature={
        'image': tf.train.Feature(
          bytes_list=tf.train.BytesList(value=[image.tobytes()])),
        'label': tf.train.Feature(
          int64_list=tf.train.Int64List(value=[label.astype(np.int64)]))
      }))
    return example

  utils.parallel_record_writer(
    _image_iterator(split), _create_example, get_split(split))

Пример #2

Файл: mnist.py Проект: ArnaudStevins/EffectiveTensorflow

def _convert_data(split):
  """Convert the dataset to TFRecord format."""
  def _create_example(item):
    image, label = item
    example = tf.train.Example(features=tf.train.Features(
      feature={
        'image': tf.train.Feature(
          bytes_list=tf.train.BytesList(value=[image])),
        'label': tf.train.Feature(
          int64_list=tf.train.Int64List(value=[label.astype(np.int64)]))
      }))
    return example

  utils.parallel_record_writer(
    _image_iterator(split), _create_example, get_split(split))