这是tensorflow2.0环境下读取tfrecords文件里数据的方法
总体分为四步:
1.加载tfrecords文件 tf.data.TFRecordDataset()
括号里填文件路径
2.创建描述功能字典
3.定义映射函数
4.通过map函数读取
如果有问题请下方留言
from __future__ import absolute_import, division, print_function, unicode_literals
from io import BytesIO
# TensorFlow and tf.keras
import tensorflow as tf
from tensorflow import keras
# Helper libraries
import numpy as np
import matplotlib.pyplot as plt
# 加载tfrecords文件
object_datasets = tf.data.TFRecordDataset("C:/Users/53111/Desktop/voc2007/JPEGImages/train.tfrecord")
# 创建一个描述功能的字典,就是你在数据里都加了什么类型的字段
# FixedLenFeature代表固定长度字段里边的中括号是必须有的
# VarLenFeature代表可变长度的字段里边只要指定类型就好
object_feature = {
'image/height'