fix typo
Browse files
utils.py
CHANGED
@@ -9,6 +9,6 @@ def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=
|
|
9 |
|
10 |
def genera(gan, batch_size):
|
11 |
with torch.no_grad():
|
12 |
-
ims = gan.G(torch.randn(batch_size, gan
|
13 |
ims = ims.permute(0,2,3,1).deatch().cpu().numpy().astype(np.uint8)
|
14 |
return ims
|
|
|
9 |
|
10 |
def genera(gan, batch_size):
|
11 |
with torch.no_grad():
|
12 |
+
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0) * 255
|
13 |
ims = ims.permute(0,2,3,1).deatch().cpu().numpy().astype(np.uint8)
|
14 |
return ims
|