(
self,
latent,
noise,
)
| 62 | return latent, noise |
| 63 | |
| 64 | def generate( |
| 65 | self, |
| 66 | latent, |
| 67 | noise, |
| 68 | ): |
| 69 | out = self.input(latent) |
| 70 | out = self.conv1(out, latent[:, 0], noise=noise[0]) |
| 71 | |
| 72 | skip = self.to_rgb1(out, latent[:, 1]) |
| 73 | i = 1 |
| 74 | for conv1, conv2, noise1, noise2, to_rgb in zip( |
| 75 | self.convs[::2], self.convs[1::2], noise[1::2], noise[2::2], self.to_rgbs |
| 76 | ): |
| 77 | out = conv1(out, latent[:, i], noise=noise1) |
| 78 | out = conv2(out, latent[:, i + 1], noise=noise2) |
| 79 | skip = to_rgb(out, latent[:, i + 2], skip) |
| 80 | if out.shape[-1] == 256: F = out |
| 81 | i += 2 |
| 82 | |
| 83 | image = skip |
| 84 | F = FF.interpolate(F, image.shape[-2:], mode='bilinear') |
| 85 | return image, F |
| 86 | |
| 87 | |
| 88 | def stylegan2( |
no outgoing calls
no test coverage detected