From d4b269d095b57ab70525212d2dd891822fbb6d46 Mon Sep 17 00:00:00 2001 From: hesuicong Date: Wed, 5 Nov 2025 15:35:33 +0800 Subject: [PATCH] =?UTF-8?q?=E9=81=AE=E6=8C=A1=E7=A8=8B=E5=BA=8F=E8=BE=93?= =?UTF-8?q?=E5=85=A5=E5=8F=82=E6=95=B0=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- libs/MVS/mask_face_occlusion.py | 21 ++++++--------------- 1 file changed, 6 insertions(+), 15 deletions(-) diff --git a/libs/MVS/mask_face_occlusion.py b/libs/MVS/mask_face_occlusion.py index b7c53fe..0af59e8 100755 --- a/libs/MVS/mask_face_occlusion.py +++ b/libs/MVS/mask_face_occlusion.py @@ -23,26 +23,21 @@ class ModelProcessor: # argv = sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else [] parser = argparse.ArgumentParser() - parser.add_argument( - "--id", - required=True, - ) - #""" parser.add_argument( - "--base_path", + "--mesh_path", type=str, required=True, ) - + parser.add_argument( - "--mesh_path", + "--sparse_dir", type=str, required=True, ) - + parser.add_argument( - "--sparse_dir", + "--output_path", type=str, required=True, ) @@ -51,12 +46,9 @@ class ModelProcessor: # print("ModelProcessor Init", args.input_file, self.pose_path) args = parser.parse_args() - self.id = args.id - - self.asset_dir = args.base_path self.mesh_path = args.mesh_path - # self.pose_path = f"{self.asset_dir}/sparse/" self.pose_path = args.sparse_dir + self.asset_dir = args.output_path if not os.path.exists(self.pose_path): raise FileNotFoundError(f"Camera data not found: {self.pose_path}") @@ -68,7 +60,6 @@ class ModelProcessor: def load_model(self): """加载并初始化3D模型""" - # model_path = f"{self.asset_dir}/repair.ply" model_path = self.mesh_path if not os.path.exists(model_path): raise FileNotFoundError(f"Model file not found: {model_path}")