CycleGAN icon indicating copy to clipboard operation
CycleGAN copied to clipboard

How could I add extra pre-processing operations only to the input of Discriminator?

Open HJLYU0519 opened this issue 5 years ago • 1 comments

I made some changes to the Discriminator. And I want to add some extra pre-processing operations only to the input of Discriminator. What should I change? And I think there are no preprocessing operations. my reason is this: in data/init.py 75 self.dataloader = torch.utils.data.DataLoader( 76 self.dataset, 77 batch_size=opt.batch_size, 78 shuffle=not opt.serial_batches, 79 num_workers=int(opt.num_threads))

Should I establish another dataloader only for Discriminator? Many thanks for your kind help

HJLYU0519 avatar Jan 23 '19 13:01 HJLYU0519

That is one possibility. You can also manually add some preprocessing code before you call the netD.forward. Like here.

junyanz avatar Jan 23 '19 13:01 junyanz