python imgtotfrecord

最新推荐文章于 2024-07-15 11:20:14 发布

Lostboy船长

最新推荐文章于 2024-07-15 11:20:14 发布

阅读量200

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/www5256246/article/details/80280391

版权

python 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

import tensorflow as tf
import glob
import os
from PIL import Image
def _int64_feature(value):
    return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]))
def _bytes_feature(value):
    return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))
def img_to_tfrecord(data_path):
    rows = 20
    cols = 20
    depth = 3
    writer = tf.python_io.TFRecordWriter('test.tfrecords')
    labelfile = open("list.txt")
    lines = labelfile.readlines()
    for line in lines:
        # print line
        #img_name = line.split(" ")[0]  # name
       # label = line.split(" ")[1]  # label
        img_path =line.strip()
        print(img_path)

        img = Image.open(img_path) #type:Image.Image

        img = img.resize((rows, cols))
        # img_raw = img.tostring()

        img_raw = img.tobytes()
        example = tf.train.Example(features=tf.train.Features(feature={
            'height': _int64_feature(rows),
            'weight': _int64_feature(cols),
            'depth': _int64_feature(depth),
            'imagebytes': _bytes_feature(img_raw),
            # 'label': _bytes_feature(label)
           }))

        writer.write(example.SerializeToString())


    writer.close()

if __name__ == '__main__':

    current_dir = os.getcwd()
    print(current_dir)
    data_path = current_dir + '/smallFace/'
    # name = current_dir + '/data'
    print('Convert start')
    img_to_tfrecord(data_path)
    print('done!')

for serialized_example in tf.python_io.tf_record_iterator("test.tfrecords"):
        example = tf.train.Example()
        example.ParseFromString(serialized_example)

        image = example.features.feature['imagebytes'].bytes_list.value

        # 可以做一些预处理之类的
        print(image)

print("-------")