push

1 files changed, 4 insertions, 3 deletions

diff --git a/test.py b/test.py
index 33edb61..97b91c7 100644
--- a/test.py
+++ b/test.py
@@ -63,8 +63,9 @@ if __name__ == '__main__':
             if last_im is not None:
                 frac = 4.
                 tmp_im = im.copy()
-                array_a = np.multiply(im, (1/frac, 1/frac, 1/frac,))
-                array_b = np.multiply(last_im, ((frac-1)/frac, (frac-1)/frac, (frac-1)/frac,))
+                array_a = np.multiply(im, 1/frac)
+                array_b = np.multiply(last_im, (frac-1)/frac)
+                print(im.shape, array_a.shape, array_b.shape)
                 im = np.add(array_a, array_b).clip(0, 255)
                 last_im = tmp_im
             else:
@@ -83,7 +84,7 @@ if __name__ == '__main__':
 
     print("________")
     print("\n")
-    
+
     cmd = ("scp", render_dir + opt.name + "_" + opt.experiment + ".mp4", "jules@asdf.us:asdf/neural/")
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()