每当我运行Tf.map时,即使使用的功能与Tf.Data教程中使用的功能非常相似,也会出现错误: self._traceback = tf_stack.extract_stack()
代码:
import numpy as np
import glob
data = "*data\\*training\\*\\*"
filelist = tf.data.Dataset.list_files(str(data))
classes = ('item0','item1','item2','item3')
def generator(file):
f = tf.io.read_file(file)
f = tf.image.decode_jpeg(f)
f = tf.image.rgb_to_grayscale(tf.image.resize(f,size = [28,28,3]))
l = file.split('\\')
l = l[2]
l_label = classes.index(l)
r1 = tf.constant(l_label)
r2 = f
return r1,r2
def runmodel():
dataset = filelist.map(todataset)
def todataset(file):
with tf.compat.v1.Session() as session:
file = file.eval(session=session)
print("I got here" + file)
session.close()
push_to_array1,push_to_array2 = generator(file.eval(session=session))
return(push_to_array2,push_to_array1)
runmodel()