bug fix for generating only 1 image

image_generator.py

@@ -117,6 +117,7 @@ def generate_images(
 
     #imgs_row = np.array()
     #imgs_complete = np.array()
+    print(seeds)
     for seed_idx, seed in enumerate(seeds):
         print('Generating image for seed %d (%d/%d) ...' % (seed, seed_idx, len(seeds)))
         z = torch.from_numpy(np.random.RandomState(seed).randn(1, G.z_dim)).to(device).float()
@@ -131,18 +132,19 @@ def generate_images(
         img = G(z, label, truncation_psi=truncation_psi, noise_mode=noise_mode)
         img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)
         print(seed_idx)
+        """
         #first image
         if seed_idx == 0:
             imgs_row = img[0].cpu().numpy()
         else:
-            imgs_row = np.hstack((imgs_row, img[0].cpu().numpy()))
+            imgs_row = np.hstack((imgs_row, img[0].cpu().numpy()))"""
         #img = PIL.Image.fromarray(img[0].cpu().numpy(), 'RGB')
         #PIL.Image.fromarray(img[0].cpu().numpy(), 'RGB').save(f'{outdir}/seed{seed:04d}.png')
     #napravi vsplit i toe to ka
-    imgs_complete = np.vstack(np.hsplit(imgs_row, 4))
+    #imgs_complete = np.vstack(np.hsplit(imgs_row, 4))
     #cv2.imshow("lalaxd", imgs_complete)
     #cv2.waitKey()
-    return PIL.Image.fromarray(imgs_complete, 'RGB')
+    return PIL.Image.fromarray(img, 'RGB')
 
 
 #----------------------------------------------------------------------------