加入快拼模式参数

main.py

@@ -112,7 +112,7 @@ class ODM_Plugin:
         """"运行OMD docker容器"""
         self.logger.info("开始运行Docker容器")
         # TODO：加一些容错处理
-        docker_runner = DockerRunner(self.project_path)
+        docker_runner = DockerRunner(self.project_path, self.config)
         docker_runner.run_odm_container()
 
     def convert_obj(self):

run.py

@@ -7,12 +7,12 @@ def parse_args():
     parser = argparse.ArgumentParser(description='ODM预处理工具')
 
     # 必需参数
-    # parser.add_argument('--image_dir',  required=True, help='输入图片目录路径')
-    # parser.add_argument('--output_dir', required=True, help='输出目录路径')
-    parser.add_argument(
-        '--image_dir', default=r'E:\datasets\UAV\134\project\images', help='输入图片目录路径')
-    parser.add_argument(
-        '--output_dir', default=r'G:\ODM_output\134', help='输出目录路径')
+    parser.add_argument('--image_dir',  required=True, help='输入图片目录路径')
+    parser.add_argument('--output_dir', required=True, help='输出目录路径')
+    # parser.add_argument(
+    #     '--image_dir', default=r'E:\datasets\UAV\134\project\images', help='输入图片目录路径')
+    # parser.add_argument(
+    #     '--output_dir', default=r'G:\ODM_output\134', help='输出目录路径')
     # 可选参数
     parser.add_argument('--mode', default='快拼模式',
                         choices=['快拼模式', '三维模式'], help='处理模式')

utils/docker_runner.py

@@ -5,7 +5,7 @@ from collections import deque
 
 
 class DockerRunner:
-    def __init__(self, project_path: str):
+    def __init__(self, project_path: str, config: dict):
         """
         初始化 DockerRunner
 
@@ -13,6 +13,7 @@ class DockerRunner:
             project_path (str): 项目路径，将挂载到 Docker 容器中
         """
         self.project_path = project_path
+        self.config = config
         self.logger = logging.getLogger("UAV_Preprocess.DockerRunner")
         self.docker_client = docker.from_env()
 
@@ -40,6 +41,11 @@ class DockerRunner:
                 "--rerun-all"
             ]
 
+            if self.config.mode == "快拼模式":
+                command += ["--fast-orthophoto", "--skip-3dmodel"]
+            else:
+                command += ["--dsm", "--dtm"]
+
             # 运行容器
             container = self.docker_client.containers.run(
                 image="opendronemap/odm:gpu",