自己用到的TensorFlow的几种输入

本文列举了自己在最近用到过的几种TensorFlow输入方式,包括从原图读取,从CSV (TXT)读取,以及从bin file读取。


1, 从原图读取。

很多情况下我们的图片训练集就是原始图片本身,并没有像cifar dataset那样存成bin等格式。因此我们需要根据一个train_list列表,去挨个读取图片。这里我用到的方法是首先将train_list.txt中的image list(也就是每一行有图片的路劲和label组成)读入队列中,那么对每次dequeue的内容中可以提取当前图片的路劲和label

filename = os.path.join(data_dir, trainfilename)
    with open(filename) as fid:
        content = fid.read()
    content = content.split('\n')
    content = content[:-1]
    valuequeue = tf.train.string_input_producer(content,shuffle=True)
    value = valuequeue.dequeue()
    dir, labels = tf.decode_csv(records=value, record_defaults=[["string"], [""]], field_delim=" ")
    labels = tf.string_to_number(labels, tf.int32)

然后从路劲dir中用tf.read_file进行读取,对读到的内容用tf.image.decode_png(或者tf.image.decode_jpeg)进行解析&#
  • 3
    点赞
  • 34
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值