配置文件处理

libs/MVS/mask_face_occlusion.py

@@ -21,27 +21,45 @@ class ModelProcessor:
             required=True, 
         )
 
-        # parser.add_argument(
-        # "--mask_image", 
-        # type=str,
-        # default="74_8" # 74_8 123_8 75_8 72_8 44_8 104_8 96_8
-        # )
+        #"""
+        parser.add_argument(
+            "--base_path", 
+            type=str,
+            required=True, 
+        )
 
+        parser.add_argument(
+            "--mesh_path", 
+            type=str,
+            required=True, 
+        )
+        
+        parser.add_argument(
+            "--sparse_dir", 
+            type=str,
+            required=True, 
+        )
+        #"""
+
+        # print("ModelProcessor Init", args.input_file, self.pose_path)
         args = parser.parse_args()
 
         self.id = args.id
-        # self.mask_image = args.mask_image
 
-        self.mesh = None
-        self.asset_dir = f"/home/algo/Documents/openMVS/data/{self.id}"
-
-        self.pose_path = f"{self.asset_dir}/sparse/"
+        # self.asset_dir = f"/home/algo/Documents/openMVS/data/{self.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
         if not os.path.exists(self.pose_path):
             raise FileNotFoundError(f"Camera data not found: {self.pose_path}")
+        
+        self.mesh = None
 
     def load_model(self):
         """加载并初始化3D模型"""
-        model_path = f"{self.asset_dir}/repair.ply"
+        # 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}")