(self, net_input, feat_map)
| 213 | self.fake_image = util.tensor2im(self.single_forward(self.net_input, self.feat_map)) |
| 214 | |
| 215 | def single_forward(self, net_input, feat_map): |
| 216 | net_input = torch.cat((net_input, feat_map), dim=1) |
| 217 | fake_image = self.netG.forward(net_input) |
| 218 | |
| 219 | if fake_image.size()[0] == 1: |
| 220 | return fake_image.data[0] |
| 221 | return fake_image.data |
| 222 | |
| 223 | |
| 224 | # generate all outputs for different styles |
no test coverage detected