ocr.py 文件源码

def distorted_inputs():
  """Construct distorted input for CIFAR training using the Reader ops.

  Returns:
    images: Images. 4D tensor of [batch_size, IMAGE_SIZE, IMAGE_SIZE, 3] size.
    labels: Labels. 1D tensor of [batch_size] size.

  Raises:
    ValueError: If no data_dir
  """
  with tf.variable_scope('distorted_inputs'):
    if not FLAGS.train_data_dir:
      raise ValueError('Please supply a data_dir')
    data_dir = FLAGS.train_data_dir
    images, labels, seq_lengths = ocr_input.distorted_inputs(data_dir=data_dir,
                                                    batch_size=FLAGS.batch_size)  
    # seq_lengths = tf.Print(seq_lengths, [seq_lengths], "seq_lengths")
    return images, labels, seq_lengths