eriklindernoren / PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.
MIT License
16.41k stars 4.07k forks source link

Switched index in loading dataset? #65

Open miqbal23 opened 5 years ago

miqbal23 commented 5 years ago

Hello, I noticed that within your implementation of pix2pix, Datasets will return image in form of like this

https://github.com/eriklindernoren/PyTorch-GAN/blob/fd9f0713eadd03b09e4fa0e828d5b7b6375e76bb/implementations/pix2pix/datasets.py#L33

But when reading it again in training loop, it was written like

https://github.com/eriklindernoren/PyTorch-GAN/blob/fd9f0713eadd03b09e4fa0e828d5b7b6375e76bb/implementations/pix2pix/pix2pix.py#L127-L128

This happened multiple times within pix2pix.py when loading the image. Is this switching intentional?

nixczhou commented 3 years ago

it is normal as the facades datasets is transfering B to A. Or You can interpret imgB = input , and imgA = output