有没有把图片标签等(如KITTI,coco数据集)转成TFRecord的代码脚本

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

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

回答 1
1 个回答
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
宁静致远 回复 Nichlas Zhaosier 1个月前 回复
<html><body><p>这个小demo挺实用,谢谢!</p></body></html>
需要登录后才能回答,快点击 来分享你的回答吧~
1 关注
0 收藏,52 浏览
邀请回答
分享到