如何解决如何使用Keras和发电机进行预测
我有一个自定义生成器,已成功用于训练Keras模型。我试图对其进行修改以用于预测:
def image_gen(ws):
filenames = glob(f'{ws}/*.h5')
for filename in filenames:
X = process(filename)
yield X
我这样调用model.predict:
filenames = glob(f'{ws}/run/*.h5')
i = 0
while True:
for results in model.predict(image_gen(ws=f'{ws}/run'),callbacks=callbacks,workers=1,steps=10,verbose=1):
np.save(filenames[i]),results)
i += 1
但是,这只能一次又一次地预测同一组10张图像。没有while true:
循环,我只得到前10张图像,然后完全停止。
如何获得model.predict
来处理我的所有文件?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。