integrate gif and gan

WiggleResults/split.py

@@ -7,7 +7,9 @@ parser.add_argument('--dim', default=128, type=int, help='dimention image')
 args = parser.parse_args()
 
 path = "."
-dirs = os.listdir(path)
+ROOT_DIR = os.path.dirname(__file__)
+print(ROOT_DIR)
+dirs = os.listdir(ROOT_DIR)
 
 dim = args.dim
 
@@ -53,12 +55,13 @@ def gif_order (data, center=True):
     #gsdfgsfgf
     return gif
 
-
+print("init gif generations!")
 # This would print all the files and directories
 for file in dirs:
+    print(file)
     if ".jpg" in file or ".png" in file:
         rowImages = []
-        im = Image.open("./" + file)
+        im = Image.open("./" + ROOT_DIR + '/' + file)
         width, height = im.size
         im = im.convert('RGB')
 
@@ -75,8 +78,9 @@ for file in dirs:
                 pointleft+= dim+4
             # Ya tengo todas las imagenes podria hacer el gif aca
             rowImages = gif_order(rowImages,center=False)
-            name = file[:-4] + "_" + str(i) + '.gif'
+            name = "./" + ROOT_DIR + '/' + file[:-4] + "_" + str(i) + '.gif'
             rowImages[0].save(name, save_all=True,format='GIF', append_images=rowImages[1:], optimize=True, duration=100, loop=0)
+            print('saved gif '+ name)
             pointtop += dim + 4
             pointleft = 3
             rowImages = []
@@ -89,3 +93,4 @@ for file in dirs:
 
     # Deleted
     #os.remove("data/" + file)
+print("end gif generations!")

app.py

@@ -55,14 +55,15 @@ def calculate_depth(model_type, img):
 
 def wiggle_effect(slider):
 
+    dim = 256 
+    c_images = 1
 
-    # python main.py --gan_type WiggleGAN --expandGen 4 --expandDis 4 --batch_size 1 --visdom false --wiggleDepth 1 --seedLoad 31219_110 --gpu_mode false --cIm 1 --imageDim 256 --name_wiggle testing_name
-    #subprocess.run([".\Scripts\activate"])
-    subprocess.run(["python", "main.py", "--gan_type", 'WiggleGAN', "--expandGen", "4", "--expandDis", "4", "--batch_size", "1", "--cIm", "1",
-                    "--visdom", "false", "--wiggleDepth", "1", "--seedLoad", '31219_110', "--gpu_mode", "false", "--imageDim", "256", "--name_wiggle", "0"
+    subprocess.run(["python", "main.py", "--gan_type", 'WiggleGAN', "--expandGen", "4", "--expandDis", "4", "--batch_size", c_images, "--cIm", c_images,
+                    "--visdom", "false", "--wiggleDepth", slider, "--seedLoad", '31219_110', "--gpu_mode", "false", "--imageDim", dim, "--name_wiggle", "result"
                     ])
-
-    return [f'Images/Input-Test/1_d.png',f'Images/Input-Test/1_d.png']
+    subprocess.run(["python",  "WiggleResults\split.py", "--dim", dim])
+    
+    return [f'WiggleResults/result.png',f'WiggleResults/result_0.gif']
 
 
 with gr.Blocks() as demo: