Upload rendering.py

visualise/rendering.py

@@ -161,14 +161,14 @@ class RenderTool():
         tmp_video_file.close()
         if int(cv2.__version__[0]) < 3:
             print('cv2 < 3')
-            writer = cv2.VideoWriter(tmp_video_file.name, cv2.cv.CV_FOURCC(*'mp4v'), 30, (width, width), True)
+            writer = cv2.VideoWriter(tmp_video_file.name, cv2.cv.CV_FOURCC(*'mp4v'), 30, (width, height), True)
         else:
             print('cv2 >= 3')
             writer = cv2.VideoWriter(tmp_video_file.name, cv2.VideoWriter_fourcc(*'mp4v'), 30, (width, height), True)
 
         center = np.mean(v_list[0][0], axis=0)
 
-        r = pyrender.OffscreenRenderer(viewport_width=800, viewport_height=800)
+        r = pyrender.OffscreenRenderer(viewport_width=800, viewport_height=viewport_height)
 
         # random exchange the position of our method and SG3D
         # pos = random.randint(1, 2)