在tensorflow的dataset.map中调用的函数中未获取张量的值

问题描述

我正在尝试从map函数调用的函数中获取张量值。但是我遇到了这个错误。

tensorflow.python.framework.errors_impl.InvalidArgumentError：您必须使用dtype字符串输入占位符张量'args_0'的值 [[{{node args_0}}]]

我的代码如下

在我的主要功能中：

# created instance of Preprocess class
preprocess = Preprocessor(is_train,TOTAL_CLASSES,OUTPUT_SHAPE)

# all the tfrecords which is present in ./tfrecords folder we will gate
dataset = tf.data.Dataset.list_files("./tfrecords")

# we will comprises it into one
dataset = tf.data.TFRecordDataset(dataset)

# Here I am mapped __call__ method which is present in Preprocess class
dataset = dataset.map(preprocess,num_parallel_calls=tf.data.experimental.AUTOTUNE)

现在在预处理类调用方法中，我遇到了错误这是我的预处理课。在通话功能中，我遇到了错误。

class Preprocessor(object):
    def __init__(self,is_train,num_classes,output_shape=(416,416)):
        self.is_train = is_train
        self.num_classes = num_classes
        self.output_shape = output_shape

    def __call__(self,example):
        features = self.parse_tfexample(example)
        encoded = features['image/encoded']
        image = tf.io.decode_jpeg(encoded)
        image = tf.cast(image,tf.float32)

        sess = tf.Session()

        # at this line I am getting error
        print(sess.run(image))

如何解决此问题。如果我对print（sess.run（image））进行注释，则此行的所有代码都可以正常工作。但是我想要映射函数中的张量值。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

deep-learning python tensorflow tf.data.dataset tfrecord