Pix2Pix模型的direction与配置文件不一致，恒定从B到A

PaddlePaddle / PaddleGAN

PaddlePaddle GAN library, including lots of interesting applications like First-Order motion transfer, Wav2Lip, picture repair, image editing, photo2cartoon, image style transfer, GPEN, and so on.

Apache License 2.0

7.9k stars 1.24k forks source link

Pix2Pix模型的direction与配置文件不一致，恒定从B到A #835

Open SageMik opened 9 months ago

SageMik commented 9 months ago

Pix2Pix模型用“AtoB”这种格式判断方向： https://github.com/PaddlePaddle/PaddleGAN/blob/21ee373c2a9481231f50b32e0753d3f543a86396/ppgan/models/pix2pix_model.py#L81 但配置文件里direction的格式却是“a2b”“b2a”： https://github.com/PaddlePaddle/PaddleGAN/blob/21ee373c2a9481231f50b32e0753d3f543a86396/configs/pix2pix_cityscapes.yaml#L20 这会导致Pix2Pix的方向恒定从B到A，一点小bug，希望修正下。

LokeZhou commented 9 months ago

感谢！如有兴趣，可修改后，提个pr