junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch
Other
23.09k stars 6.32k forks source link

Training the model on custom dataset #1613

Open AGRocky opened 1 year ago

AGRocky commented 1 year ago

HI team it's a wonderful work you've. It's appreciable. However I am trying to fine tune the model on my custom dataset which has noise(check boxes) in them and need the output as data without checkboxes. I trained the model using the necessary requirements of the CycleGAN but still when I test the the images with checkboxes I am getting blank output.

I humbly request anyone form the team or from the community help me out with this, It would deeply appreciated.