gpt4 book ai didi

python - Keras 自定义生成器 TypeError : 'NoneType' object is not callable

转载 作者:行者123 更新时间:2023-11-28 18:06:14 25 4
gpt4 key购买 nike

我编写了一个自定义生成器来加载我的数据集并将其提供给 fit_generator 方法。但我收到一个错误。 question_trainimg_lis_trainanswers_train 是字符串列表。我希望 mygen 返回 32 个格式样本的批处理:

[images,questions] , answers

代码如下:

def mygen(questions_train,img_lis_train,answers_train):
start = 0
data_size = len(questions_train)
batch_size = 32
while True:
if( start+batch_size <= data_size ):
batch_ques = questions_train[ start : start+batch_size ]
batch_ans = answers_train[ start : start+batch_size ]
batch_img_names = img_lis_train[ start : start+batch_size ]
elif(start < data_size):
batch_ques = questions_train[ start : ]
batch_ans = answers_train[ start : ]
batch_img_names = img_lis_train[ start : start+batch_size ]
else:
start = 0
continue

batch_img = []
for img_name in batch_img_names:
img = load_img('./dataset/images/' + str(img_name) + '.png' , target_size = (224,224))
img = img_to_array(img)
batch_img.append( preprocess_input(img) )

start += 32
print('start = ' + str(start))
yield [batch_img, batch_ques] ,batch_ans

fc_model.fit_generator(mygen, steps_per_epoch=3376, epochs = 10)

这是我得到的错误:

  File "mycode.py", line 210, in <module>
fc_model.fit_generator(mygen, steps_per_epoch=3376, epochs = 10)
File "/opt/apps/Python-3.5.1/lib/python3.5/site-packages/keras/legacy/interfaces.py", line 87, in wrapper
return func(*args, **kwargs)
File "/opt/apps/Python-3.5.1/lib/python3.5/site-packages/keras/models.py", line 1223, in fit_generator
initial_epoch=initial_epoch)
File "/opt/apps/Python-3.5.1/lib/python3.5/site-packages/keras/legacy/interfaces.py", line 87, in wrapper
return func(*args, **kwargs)
File "/opt/apps/Python-3.5.1/lib/python3.5/site-packages/keras/engine/training.py", line 2083, in fit_generator
generator_output = next(output_generator)
StopIteration
Exception ignored in: <bound method BaseSession.__del__ of <tensorflow.python.client.session.Session object at 0x7f936527fcc0>>
Traceback (most recent call last):
File "/opt/apps/Python-3.5.1/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 712, in __del__
File "/opt/apps/Python-3.5.1/lib/python3.5/site-packages/tensorflow/python/framework/c_api_util.py", line 31, in __init__
TypeError: 'NoneType' object is not callable

最佳答案

您需要通过调用生成器并将数据传递给它来构造生成器,否则它应该如何生成批处理?这是正确的做法:

fc_model.fit_generator(mygen(q_train, i_train, a_train), ...)

关于python - Keras 自定义生成器 TypeError : 'NoneType' object is not callable,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53277465/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com