tensorflow如何批量读取图片

 更新时间:2019年08月29日 17:23:07   作者:雪儿waii  
这篇文章主要为大家详细介绍了tensorflow如何批量读取图片,具有一定的参考价值,感兴趣的小伙伴们可以参考一下

本文实例为大家分享了tensorflow如何批量读取图片的具体代码,供大家参考,具体内容如下

代码:

import tensorflow as tf
import os


def picread(filelist):
 """
 读取狗的图片并转换成张量
 :param filelist: 文件路f径+名字的列表
 :return: 每张图片的张量
 """
 # 1.构造文件的队列
 file_queue = tf.train.string_input_producer(filelist)

 # 2.构造阅读器去读取图片内容(默认读取一张图片)
 reader = tf.WholeFileReader()
 key,value = reader.read(file_queue)

 # 3.对读取的图片进行解码
 image = tf.image.decode_jpeg(value)

 # 4.处理图片的大小(统一大小)
 image_resize = tf.image.resize_images(image,[200,200])

 # 注意:一定要把样本的形状固定,在批处理中要求所有数据的形状必须固定
 image_resize.set_shape([200,200,3])


 # 5.进行批处理
 image_resize_batch = tf.train.batch([image_resize],batch_size=3,num_threads=1,capacity=3)


 return image_resize


#批处理大小,跟队列,数据的数量没有影响,只决定 这批次处理多少数据

if __name__ == "__main__":
 # 1.找到文件,放入列表 路径+名字 ->列表当中
 file_name = os.listdir("./data/dogpic/")

 filelist = [os.path.join("./data/dogpic/",file) for file in file_name ]
 image_batch= picread(filelist)

 #开启会话运行结果
 with tf.Session() as sess:
  #定义一个线程协调器
  coord = tf.train.Coordinator()

  #开启读文件的线程
  threads = tf.train.start_queue_runners(sess,coord=coord)

  #打印读取的内容
  print(sess.run([image_batch]))

  #回收子线程
  coord.request_stop()
  coord.join(threads)

结果:

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

相关文章

  • Python 线程池模块之多线程操作代码

    Python 线程池模块之多线程操作代码

    最近在做一个爬虫相关的项目,单线程的整站爬虫,耗时真的不是一般的巨大,运行一次也是心累,所以,要想实现整站爬虫,多线程是不可避免的,那么python多线程又应该怎样实现呢?今天小编给大家分享下实现代码,感兴趣的朋友一起看看吧
    2021-05-05
  • 基于OpenCv实现的人脸识别(附Python完整代码)

    基于OpenCv实现的人脸识别(附Python完整代码)

    人脸识别是基于人的脸部特征信息进行身份识别的一种生物识别技术,下面这篇文章主要给大家介绍了关于如何基于OpenCv实现的人脸识别,文中还附Python完整代码,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2022-11-11
  • django使用graphql的实例

    django使用graphql的实例

    这篇文章主要介绍了django使用graphql的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-09-09
  • 详解Python sys.argv使用方法

    详解Python sys.argv使用方法

    在本文中我们给大家详细讲解了关于Python sys.argv使用方法以及注意事项,有此需要的读者们跟着学习下。
    2019-05-05
  • Python的getattr函数方法学习使用示例

    Python的getattr函数方法学习使用示例

    这篇文章主要为大家介绍了Python的getattr方法学习使用示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-08-08
  • python线程池 ThreadPoolExecutor 的用法示例

    python线程池 ThreadPoolExecutor 的用法示例

    这篇文章主要介绍了python线程池 ThreadPoolExecutor 的用法示例,帮助大家更好得理解和使用python线程池,感兴趣的朋友可以了解下
    2020-10-10
  • python读写csv文件方法详细总结

    python读写csv文件方法详细总结

    在本文中小编给各位分享的是关于python读写csv文件方法的详细内容,对此有需要的朋友们跟着学习参考下。
    2019-07-07
  • 用python实现五子棋实例

    用python实现五子棋实例

    这篇文章主要为大家详细介绍了用python实现五子棋实例,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-05-05
  • 详解pycharm自动import所需的库的操作方法

    详解pycharm自动import所需的库的操作方法

    这篇文章主要介绍了pycharm自动import所需的库的操作方法,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-11-11
  • 详解用Python处理HTML转义字符的5种方式

    详解用Python处理HTML转义字符的5种方式

    本文介绍了详解用Python处理HTML转义字符的5种方式,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-12-12

最新评论