Keras - How are batches and epochs used in fit_generator()?

后端 未结 3 570
梦谈多话
梦谈多话 2020-12-31 02:34

I have a video of 8000 frames, and I\'d like to train a Keras model on batches of 200 frames each. I have a frame generator that loops through the video frame-by-frame and a

3条回答
  •  执念已碎
    2020-12-31 03:05

    You can force your generator to reset itself by adding a while 1: loop, that's how I proceed. Thus your generator can yield batched data for each epochs.

提交回复
热议问题