有没有把图片标签等（如KITTI，coco数据集）转成TFRecord的代码脚本

KITTI数据集 TensorFlow

宁静致远提问于 2年前

近期在做一个KITTI数据集相关的训练，但发现GitHub上提供的KITTI_to_TFRecord的脚本跑不通，一直报错。

麻烦问下上述问题如何解决。不胜感激。

赞 2 回答 1

1 个回答

Nichlas Zhaosier

2年前

from PIL import Image
import tensorflow as tf
import os

tfr_out = "t.tfr"
wr = tf.python_io.TFRecordWriter(tfr_out)


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]))


_id = 1

for img_name in os.listdir("./img"):
    _id = _id + 1
    print(img_name)
    img = Image.open("./img/" + img_name)
    img = img.resize((800, 800))
    img_raw = img.tobytes()
    feature = {"label": _int64_feature(_id), "img_raw": _bytes_feature(tf.compat.as_bytes(img_raw))}
    example = tf.train.Example(features=tf.train.Features(feature=feature))
    wr.write(example.SerializeToString())

图片转TFRecord，你可以试试。

赞 1 回复 · 1

宁静致远回复 Nichlas Zhaosier 2年前回复

这个小demo挺实用，谢谢！

需要登录后才能回答，快点击或来分享你的回答吧~