junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch
Other
22.8k stars 6.29k forks source link

Adding New Generator, new Discriminator, and new Loss in Pix2Pix Model #1531

Open hendrikTpl opened 1 year ago

hendrikTpl commented 1 year ago

hi, I am trying to do image editing based on Pix2Pix model, is there any guide which part of the code that i need to take care beside network.py, pix2pix_model.py if I want to add new Network Model for both Generator and Discriminator as well as the respective loss ?