TensorFlow之tf.train.slice_input_producer读取数据到队列

最新推荐文章于 2020-08-01 16:36:16 发布

kk123k

最新推荐文章于 2020-08-01 16:36:16 发布

阅读量2.7k

点赞数 2

分类专栏： TensorFlow 文章标签： tf.train.slice_input_producer python tensorflow 深度学习 tf.train.string_input_producer

本文链接：https://blog.youkuaiyun.com/kk123k/article/details/86772813

版权

tf.train.slice_input_producer被定义于tensorflow/python/training/input.py，用于定义读取数据到队列中的方式。

1、tf.train.slice_input_producer和tf.train.string_input_producer的区别

注意tf.train.slice_input_producer和tf.train.string_input_producer函数的区别：

tf.train.string_input_producer只能输入张量的列表，而tf.train.slice_input_producer可以将image_list，label_list等多个list组合成一个tensor_list作为输入。如下：

    image_dir ='path/to/image_dir/*.jpg'
    image_list = glob.glob(image_dir)
    label_list=...
    image_list = tf.convert_to_tensor(image_list, dtype=tf.string)
    # 可以将image_list,label_list多个list组合成一个tensor_list
    image_que, label_que = tf.train.slice_input_producer([image_list,label_list], num_epoc