近期在做一个KITTI数据集相关的训练,但发现GitHub上提供的KITTI_to_TFRecord的脚本跑不通,一直报错。
麻烦问下上述问题如何解决。不胜感激。
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,你可以试试。
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,你可以试试。