(self)
| 45 | self.bg = self.real * (1 - self.mask) |
| 46 | |
| 47 | def forward(self): |
| 48 | self.output = self.netG(self.inputs, self.mask) |
| 49 | self.fake_f = self.output * self.mask |
| 50 | self.attentioned = self.output * self.mask + self.inputs[:,:3,:,:] * (1 - self.mask) |
| 51 | self.harmonized = self.attentioned |
| 52 | |
| 53 | def backward_G(self): |
| 54 | """Calculate GAN and L1 loss for the generator""" |