我可以将解码后的图像作为功能写入Tensorflow tfrecord，然后读取它吗？

import tensorflow as tf

# 读取解码后的图像
image = tf.io.decode_jpeg(tf.io.read_file('image.jpg'))

# 创建tfrecord样本
example = tf.train.Example(features=tf.train.Features(feature={
    'image': tf.train.Feature(bytes_list=tf.train.BytesList(value=[tf.io.encode_jpeg(image).numpy()]))
}))

# 将tfrecord样本写入文件
with tf.io.TFRecordWriter('data.tfrecord') as writer:
    writer.write(example.SerializeToString())

import tensorflow as tf

# 定义tfrecord文件路径
tfrecord_file = 'data.tfrecord'

# 定义解析函数
def parse_example(example):
    features = tf.io.parse_single_example(example, features={
        'image': tf.io.FixedLenFeature([], tf.string)
    })
    image = tf.io.decode_jpeg(features['image'])
    return image

# 读取tfrecord文件
dataset = tf.data.TFRecordDataset(tfrecord_file)

# 解析tfrecord样本
dataset = dataset.map(parse_example)

# 打印图像数据
for image in dataset:
    print(image)