search

tom-doerr / TecoGAN-Docker

This is a fork of the TecoGAN project (https://github.com/thunil/TecoGAN) that adds support for docker.
Apache License 2.0
105 stars 21 forks source link

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

Open alessiapacca opened 4 years ago

alessiapacca commented 4 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.

tom-doerr commented 4 years ago

Don't know about that. You might want to also post it in the main TecoGAN repo where more people will see this.