Python shuffle_batch_join示例

示例#1

文件： train.py 项目： xiaoye77/Face-hallucination-with-tiny-images

def input_pipeline(glob_pattern, batch_size, num_threads, num_epochs):
    tfrecords_list = glob.glob(glob_pattern)
    # num_examples = utils.num_examples_from_tfrecords(tfrecords_list) # takes too long time for lsun
    X, sample_z = ip.shuffle_batch_join(tfrecords_list,
                                        batch_size=batch_size,
                                        num_threads=num_threads,
                                        num_epochs=num_epochs)
    return X, sample_z

示例#2

def input_pipeline(glob_pattern,
                   batch_size,
                   num_threads,
                   num_epochs,
                   image_size,
                   dataset=''):
    tfrecords_list = glob.glob(glob_pattern)
    X = ip.shuffle_batch_join(tfrecords_list,
                              batch_size=batch_size,
                              num_threads=num_threads,
                              num_epochs=num_epochs,
                              image_size=image_size,
                              dataset=dataset)
    return X

示例#3

def input_pipeline(glob_pattern, batch_size, num_threads, num_epochs):
    tfrecords_list = glob.glob(glob_pattern)
    num_examples = utils.num_examples_from_tfrecords(tfrecords_list)
    X = ip.shuffle_batch_join(tfrecords_list, batch_size=batch_size, num_threads=num_threads, num_epochs=num_epochs)
    return X, num_examples