学习笔记:tf.data对tfrecord数据进行解析,获取data_batch
import tensorflow as tf
import tensorflow.keras.backend as K
import numpy as np
def _parse_function(example_proto):
# map_func: apply to each element of this dataset
features = {
'im...
原创
2019-10-04 21:51:20 ·
646 阅读 ·
0 评论