我有一个问题，转换tfrecords回图像：

def _parse_test_image_function(img):
    image_feature_description = {
        'image/file_name': tf.io.FixedLenFeature([], tf.string),
        'image/encoded_image': tf.io.FixedLenFeature([], tf.string),
    }
    return tf.io.parse_single_example(img, image_feature_description)

test_dataset = tf.data.TFRecordDataset(temp_path)
test_dataset = test_dataset.map(_parse_test_image_function)

print(tf.__version__)
images = test_dataset.take(1)
print(images)

2.5.0
<TakeDataset shapes: {image/encoded_image: (), image/file_name: ()}, types: {image/encoded_image: tf.string, image/file_name: tf.string}>

image_feature_description中的字段正确
我也看到了这个Converting TFRecords back into JPEG Images，但这对我来说没有太大帮助，因为答案中使用的一些函数已经过时了。