thunil / TecoGAN

This repo contains source code and materials for the TEmporally COherent GAN SIGGRAPH project.
Apache License 2.0
5.95k stars 1.13k forks source link

Is there a way to get all the 128x128 patches of a single frame? #95

Open alessiapacca opened 3 years ago

alessiapacca commented 3 years ago

I was thinking about putting the 128x128 frames back together before calculating the losses. In the file Teco.py, the loss is calculated only on the single 128x128 patch of the generated image.

For example, this https://github.com/tom-doerr/TecoGAN-Docker/blob/e78b1d2f3d3e9ae00df7b5a123118f375167a712/lib/Teco.py#L168 is the s_gen_output that is afterwartds used in https://github.com/tom-doerr/TecoGAN-Docker/blob/e78b1d2f3d3e9ae00df7b5a123118f375167a712/lib/Teco.py#L320 to compute the loss. Is there a way to reach out all the patches of a single frame or not?

I assume batch = 4 means we are dealing with 4 different patches of different videos frames. And RNN_N being equal to 10 means we have a sequence of 10 patches from 10 consequent frames? So I was wondering if we can somehow obtain the output of all the 128x128 patches of a frame or not.