Update test_03_poisson.py

SangHunHan92 · Aug 2, 2023 · 5a26622 · 5a26622
1 parent 52ff2a6
commit 5a26622
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test_03_poisson.py b/test_03_poisson.py
@@ -13,12 +13,12 @@
 parser.add_argument('--save_name', type=str, default='test', help='path to save folder')
 args = parser.parse_args()
 
-def main(save_path, proj_name):
+def main(save_path, save_name):
 
-    pc_path = os.path.join(save_path, proj_name, 'output_plys')
-    obj_path = os.path.join(save_path, proj_name, 'output_objs')
-    # pc_path = os.path.join(save_path, proj_name, 'output_plys_c')
-    # obj_path = os.path.join(save_path, proj_name, 'output_objs_c')
+    pc_path = os.path.join(save_path, save_name, 'output_plys')
+    obj_path = os.path.join(save_path, save_name, 'output_objs')
+    # pc_path = os.path.join(save_path, save_name, 'output_plys_c')
+    # obj_path = os.path.join(save_path, save_name, 'output_objs_c')
     os.makedirs(obj_path, exist_ok=True)
     os.chmod(obj_path, 0o777)
 
@@ -54,5 +54,5 @@ def main(save_path, proj_name):
         ex_mesh.export(save_path)
 
 if __name__ == '__main__':    
-    main(args.save_path, args.proj_name)
+    main(args.save_path, args.save_name)