along/UAV2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

加入快拼模式参数

Browse Source
This commit is contained in:
weixin_46229132 2025-04-08 14:24:18 +08:00
parent 4c460515c5
commit e758eeffb9
3 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

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

12
run.py
View File

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

8
utils/docker_runner.py
View File

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