TensorFlow通过文件名/文件夹名获取标签,并加入队列的实现
2020-03-11 16:47
381 查看
TensorFlow官网给的cifar-10教程,是卷积神经网络入门的好例子,有时想直接拿这个模型来跑自己的数据,却发现他的数据类型不是常见的,我们一般获取的数据(图片)以文件夹分类存好,或者直接在文件名上注明了类别,这时就要通过文件名的来获取标签,显然直接用cifar-10的方式是不行的。
这里当然可以吧数据转化成cifar-10那种类型,不过个人不喜欢这种方式吧。
另一种就还是老办法,对文件名进行处理,获取标签。
其实处理文件名,对python来说,很简单。只是刚接触卷积神经网络的,可能并不熟悉TensorFlow的用法(本人),怎么修改代码才能以这种方式读取数据进队列呢?
看代码:
import tensorflow as tf import os #这就是数据所在的路径,其子目录是若干个文件夹,分别对应一类数据, #文件夹下存放的就是对应那一类的数据。 #命名方式:1,2,3,4・・・・(代表不同的类别,简单粗暴) path="./data" classes=2#数据类别的数量 imagesList=[]#存放图片数据的列表(这里存放的仅仅是图片所在的路径,并不是图片) labelsList=[]#存放标签,与上面图片是一一对应的 filepaths=[os.path.join(path,"%d"%i)for i in range (1,classes+1)]#存放path的子目录的路径 for p in filepaths: for filename in os.listdir(p):#获取图片的名字 imagesList.append(os.path.join(p,filename))#图片名和路径拼接,然后如队列 labelsList.append(int(p[-1]))#这里是按照命名的方式,p的最后一个字符就是类别 image = tf.cast(imagesList,tf.string)#转化成tf的数据类型 label = tf.cast(labelsList,tf.int64) queue = tf.train.slice_input_producer([image,label])#生成队列,这里也就关键的地方 label = queue[1] image_c = tf.read_file(queue[0]) image = tf.image.decode_jpeg(image_c,channels = 3) training_image_data = tf.image.resize_images(image, [2, 2]) #这个函数是重塑行,具体干嘛不清楚,不用有时报错 example_batch, label_batch = tf.train.shuffle_batch([training_image_data,label], batch_size=1, capacity=2, min_after_dequeue=1) #生成Batch # 运行Graph with tf.Session() as sess: coord = tf.train.Coordinator() #创建一个协调器,管理线程 threads = tf.train.start_queue_runners(coord=coord) #启动QueueRunner, 此时文件名队列已经进队。 for i in range(6): e_val,l_val = sess.run([example_batch, label_batch]) print (e_val,l_val)#弄些简单的数据来测试效果。 coord.request_stop() coord.join(threads)
以上这篇TensorFlow通过文件名/文件夹名获取标签,并加入队列的实现就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- 进击的KFC:iOS WebView 如何通过js获取网页中所有图片并加入点击事件,实现浏览图片的功能
- javascript通过获取html标签属性class实现多选项卡的方法
- Windows C++ 获取文件夹大小(通过FindNextFile实现)
- javascript通过获取html标签属性class实现多选项卡的方法
- Linux C++ 获取文件夹大小2(通过ftw实现)
- Python 从EXEL中读取文件名,并通过模糊搜索实现批量转移指定文件到指定文件夹
- iOS WebView 如何通过js获取网页中所有图片并加入点击事件,实现浏览图片的功能
- C# 实现读取本地某文件夹中的html文件,然后通过正则表达式获取input控件中的id值
- C#编程实现获取文件夹中所有文件的文件名
- BuguMongo通过BuguFS类来操作GridFS文件系统,能够实现对文件的保存、获取、删除、重命名、移动等操作,而且能够按文件夹组织文件。
- Linux C++获取文件夹大小1(通过lstat实现)
- Python 获取当前文件夹下所有文件名
- 使用C++获取文件夹中所有文件名(windows环境)
- 通过rabbitMQ消息队列实现分布式环境下的最终一致性
- PHP 通过WMI来实现获取客户端计算机硬件及系统信息:
- VC实现类似Excel文件夹式样的标签控制
- java通过ActiveMQ实现JMS的消息队列实例
- 通过摄像机视频设备或者流媒体服务器SDK获取到数据转换成RTMP流实现网页/手机微信播放
- 通过脚本实现AD用户自动连接打印机与共享文件夹
- 通过os中的os.path.basename获取路径中的文件名