XiaoCode-er / Two-Stream-CNN

Two Stream CNN implemented in Keras using in skeleton-based action recognition with dataset NTU RGB+D
76 stars 13 forks source link

您好,请问一下,倘若是送入两组处理好的图片进这个双流CNN,应该怎么修改代码呢? #9

Open simay233 opened 4 years ago

simay233 commented 4 years ago

我之前利用keras自带的Imagegenerator函数,想用Imagegenerator.flow_from_directory()来处理两组图像,然后使用model.fit.generator()函数对模型进行拟合,但是由于keras自的.fit.generator()函数固定只能输入一个generator(尝试改了源代码但是一直报错),所以未能成功。很幸运找到了这个项目,想在您的基础上修改定义一个(训练和测试分别都)送入两组图片的函数,看您用train_on_batch()可以实现多输入,我想请问一下您应该怎么根据我的实际情况来定义自己的data_generator呢?(万分感激!)