每天一个知识点(八)

作者: 少年小马 | 来源:发表于2020-04-02 08:11 被阅读0次

TensorFlow的数据读取机制:(读取图片为例)

      如果将图片先读取到内存中后提供给GPU或CPU计算,这样GPU在数据读取的时间是无事可做的,这大大降低运算效率。为此将读取数据和计算分别放在两个线程中去做。一个线程负责源源不断的将图片读取到内存的一个队列中,另一个线程直接从队列中取用计算。

      机器学习通常使用epoch来重复计算,运行一个epoch就是将数据集中的所有图片都计算一遍,两个epoch计算两遍。为了方便管理epoch,TensorFlow在内存队列前又添加了一个队列叫做‘文件名队列’。使用 tf.train.string_input_producer函数来创建文件名队列,需要传入一个文件名 list,系统会自动将它转换成一个文件名队列。此外还需要num_epoch(epoch的 数目)和shuffle(在文件名队列中打乱图片顺序)两个参数。在TensorFlow中不需要自己创建内存队列,使用reader对象从文件名中读取数据即可,如下所示。

reader = tf.WholeFileReader  key, value = reader.read(filename_queue)#filename_queue是文件名队列。

      一切准备就绪之后,咱们定义的文件名队列中并没有开始读入图片,需要使用tf.train.star_queue_runners函数进行激活。如下是读取三张图片的程序:

with tf.Session() as sess:

    filename = ['A.jpg', 'B.jpg', 'C.jpg']

    filename_queue = tf.train.string_input_producer(filename, huffle=false,num_epoch=5)

    reader = tf.WholeFileReader()

    key, value = reader.read(filename_queue)

    tf.local_variables_initializer().run()

    threads = tf.train.start_queue_runners(sess=sess)

    i = 0

    while True :

    i+=1

    image_data = sess.run(value)

    with open('read/test_%d.jpg'%i, 'wb') as f:

    f.write(image_data)

相关文章

  • 每天一个知识点(八)

    TensorFlow的数据读取机制:(读取图片为例) 如果将图片先读取到内存中后提供给GPU或CPU计算,...

  • 完整的整合营销传播套路梳理

    每天学习一个营销知识点,每天梳理营销知识点,每天分享一个营销知识点。 今天梳理下整合营销传播的整体构建框架怎么做?...

  • 每天一个知识点

    图像识别与目标检测: 输入一张图片,输出与该图片对应的类别属于图像识别。如果输出的不单单是图像类别,还包括图像中所...

  • 每天一个知识点

    Toast.LENGTH_SHORT和 Toast.LENGTH_LONG分别对应多长时间? 多次调用toast....

  • 《心理学》第11次课程--李媛媛

    ①印象最深: 1.谢婷同学讲解第八章知识点 2.老师讲解第八章知识点 3.看了一个很暖心的小视频 ②原因: 1.准...

  • 84.Kotlin object 和 companion obj

    object EightyFourActivity {/**** 每天一个知识点day84 TODO Kotlin...

  • 编程也要温故而知新

    文|一本正经胡说八道的猫 跟着教材连续学了一个星期,知识点不断刷新我的大脑,一开始很兴奋,每天都在快速地成长着,然...

  • 知识点八

    1、公共样式的提取 2、CSS2d 的转换 例子: 2.1translate位移 2.2rotate旋转 2.3 ...

  • 每天一个知识点(二)

    有时候,同样的代码源程序与交互式终端的执行结果不同。这是因为源程序是一次执行,系统只开辟一个内存空间,这让相同的数...

  • 每天一个知识点(四)

    介绍几个数据集 LFW(labeled faces in the wild)人脸数据库: 主要用来研究非受限情况下...

网友评论

    本文标题:每天一个知识点(八)

    本文链接:https://www.haomeiwen.com/subject/eijtphtx.html