Open Sean16SYSU opened 4 years ago
I don't know the details, but your discriminator might be PatchGAN, it looks at overlapping patches 70x70 and tells if this path is generated or not. So this is might be the case. Hope I've helped you :)
I have implemented DualGAN now, and found that wgan works well only when D without BN layers.
D may not suitable for any size image. And we would modify slightly before applying it for your own dataset.
My fault.
Thanks and I will shut down this issue. And I am always welcomed for further communicated for GANs model implementation with Pytorch. Anyone is free to email me.
@AverageName Thanks for your help. Actually, DualGAN ultilizes WGAN instead of PatchGAN which explained in CycleGAN. But now I feel this case reasonable, as the models may not suitable for any possible input of any size.
I have tested the models in DualGAN and got something unexpected. Given input tensor whose shape is [B, 3, 256, 256], I put it into Discriminator in DualGAN and the shape of output tensor is [B, 1, 30, 30], instead of [B, 1, 1, 1].
The code I run will be list at the bottom, most of the code is copied from
models.py
in dualgan, except for theimport
andmain
.the output of the code is shown as follows:
Code: