Rodrigo_Cobo
commited on
Commit
•
59fba04
1
Parent(s):
b8400b6
solved: the problem was in the input name
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ def calculate_depth(model_type, img):
|
|
13 |
if not os.path.exists('temp'):
|
14 |
os.system('mkdir temp')
|
15 |
|
16 |
-
filename = "Images/Input-Test/
|
17 |
|
18 |
img.save(filename, "PNG")
|
19 |
|
@@ -49,7 +49,7 @@ def calculate_depth(model_type, img):
|
|
49 |
|
50 |
formatted = (output * 255.0 / np.max(output)).astype('uint8')
|
51 |
out_im = Image.fromarray(formatted)
|
52 |
-
out_im.save("Images/Input-Test/
|
53 |
|
54 |
return out_im
|
55 |
|
@@ -57,13 +57,14 @@ def wiggle_effect(slider):
|
|
57 |
|
58 |
dim = '256'
|
59 |
c_images = '1'
|
|
|
60 |
|
61 |
subprocess.run(["python", "main.py", "--gan_type", 'WiggleGAN', "--expandGen", "4", "--expandDis", "4", "--batch_size", c_images, "--cIm", c_images,
|
62 |
-
"--visdom", "false", "--wiggleDepth", str(slider), "--seedLoad", '31219_110', "--gpu_mode", "false", "--imageDim", dim, "--name_wiggle",
|
63 |
])
|
64 |
subprocess.run(["python", "WiggleResults/split.py", "--dim", dim])
|
65 |
|
66 |
-
return [f'WiggleResults/
|
67 |
|
68 |
|
69 |
with gr.Blocks() as demo:
|
|
|
13 |
if not os.path.exists('temp'):
|
14 |
os.system('mkdir temp')
|
15 |
|
16 |
+
filename = "Images/Input-Test/1.png"
|
17 |
|
18 |
img.save(filename, "PNG")
|
19 |
|
|
|
49 |
|
50 |
formatted = (output * 255.0 / np.max(output)).astype('uint8')
|
51 |
out_im = Image.fromarray(formatted)
|
52 |
+
out_im.save("Images/Input-Test/1_d.png", "PNG")
|
53 |
|
54 |
return out_im
|
55 |
|
|
|
57 |
|
58 |
dim = '256'
|
59 |
c_images = '1'
|
60 |
+
name_output = 'out'
|
61 |
|
62 |
subprocess.run(["python", "main.py", "--gan_type", 'WiggleGAN', "--expandGen", "4", "--expandDis", "4", "--batch_size", c_images, "--cIm", c_images,
|
63 |
+
"--visdom", "false", "--wiggleDepth", str(slider), "--seedLoad", '31219_110', "--gpu_mode", "false", "--imageDim", dim, "--name_wiggle", name_output
|
64 |
])
|
65 |
subprocess.run(["python", "WiggleResults/split.py", "--dim", dim])
|
66 |
|
67 |
+
return [f'WiggleResults/'+ name_output + '.jpg',f'WiggleResults/' + name_output + '_0.gif']
|
68 |
|
69 |
|
70 |
with gr.Blocks() as demo:
|