implemented generating 16 images instead of 1

image_generator.py

@@ -138,7 +138,7 @@ def generate_images(
         else:
             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')
+        #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))
     #cv2.imshow("lalaxd", imgs_complete)