Spaces:
Runtime error
Runtime error
Commit
·
e8d2a62
1
Parent(s):
12ded9c
Update utils.py
Browse files
utils.py
CHANGED
@@ -7,7 +7,7 @@ def carga_modelo(model_name = 'ceyda/butterfly_cropped_uniq1K_512', model_versio
|
|
7 |
gan.eval()
|
8 |
return gan
|
9 |
|
10 |
-
def genera(gan, batch_size = 1)
|
11 |
with torch.no_grad() #no queremos entrenar el modelo
|
12 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp(0.0,1.0) * 255 #generamos imágenes y las aplastamos entre 0 y 1
|
13 |
ims = ims.permute(0,2,3,1).detach().cpu().numpy().astype(np.uint8) #las pasamos a numpy
|
|
|
7 |
gan.eval()
|
8 |
return gan
|
9 |
|
10 |
+
def genera(gan, batch_size = 1):
|
11 |
with torch.no_grad() #no queremos entrenar el modelo
|
12 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp(0.0,1.0) * 255 #generamos imágenes y las aplastamos entre 0 y 1
|
13 |
ims = ims.permute(0,2,3,1).detach().cpu().numpy().astype(np.uint8) #las pasamos a numpy
|