np.squeeze expects int or tuple

see:https://docs.scipy.org/doc/numpy/reference/generated/numpy.squeeze.html

demo.py

@@ -61,9 +61,9 @@ def demo():
 
 				# reshape outputs
 				input_img = np.squeeze(input_img)
-				shape = np.squeeze(face_shape,[0])
-				color = np.squeeze(face_color,[0])
-				landmarks_2d = np.squeeze(landmarks_2d,[0])
+				shape = np.squeeze(face_shape, (0))
+				color = np.squeeze(face_color, (0))
+				landmarks_2d = np.squeeze(landmarks_2d, (0))
 
 				# save output files
 				# cropped image, which is the direct input to our R-Net
@@ -73,4 +73,4 @@ def demo():
 				save_obj(os.path.join(save_path,file.split('\\')[-1].replace('.png','_mesh.obj')),shape,tri,np.clip(color,0,255)/255) # 3D reconstruction face (in canonical view)
 
 if __name__ == '__main__':
-	demo()
+	demo()