subprocess.run

This commit is contained in:
龙澳 2024-12-27 19:40:33 +08:00
parent bb17302919
commit 6ad875d59a
3 changed files with 42 additions and 38 deletions

View File

@ -307,7 +307,7 @@ class ImagePreprocessor:
if __name__ == "__main__": if __name__ == "__main__":
# 创建配置 # 创建配置
config = PreprocessConfig( config = PreprocessConfig(
image_dir=r"F:\error_data\20241024100834\project\images", image_dir=r"G:\error_data\20241024100834\project\images",
output_dir=r"G:\ODM_output\20241024100834", output_dir=r"G:\ODM_output\20241024100834",
cluster_eps=0.01, cluster_eps=0.01,

View File

@ -20,12 +20,18 @@ class ODMProcessMonitor:
def run_odm_with_monitor(self, grid_dir: str, grid_idx: int, fast_mode: bool = True) -> Tuple[bool, str]: def run_odm_with_monitor(self, grid_dir: str, grid_idx: int, fast_mode: bool = True) -> Tuple[bool, str]:
"""运行ODM命令""" """运行ODM命令"""
try:
self.logger.info(f"开始处理网格 {grid_idx + 1}") self.logger.info(f"开始处理网格 {grid_idx + 1}")
# 规范化路径
grid_dir = os.path.abspath(grid_dir).replace('\\', '/')
# 设置环境变量
# env = os.environ.copy()
# env["PYTHONUNBUFFERED"] = "1"
# 构建命令字符串 # 构建命令字符串
command = ( command = (
f"docker run -ti --rm " f"docker run --rm " # 移除 -ti 参数
f"-v {grid_dir}:/datasets " f"-v {grid_dir}:/datasets "
f"opendronemap/odm " f"opendronemap/odm "
f"--project-path /datasets project " f"--project-path /datasets project "
@ -47,8 +53,11 @@ class ODMProcessMonitor:
result = subprocess.run( result = subprocess.run(
command, command,
shell=True, shell=True,
stdout=subprocess.DEVNULL, # env=env,
stderr=subprocess.DEVNULL cwd=grid_dir, # 设置工作目录
stdout=subprocess.DEVNULL, # 完全禁用输出捕获
stderr=subprocess.DEVNULL,
creationflags=subprocess.CREATE_NO_WINDOW if os.name == 'nt' else 0 # Windows下不创建新窗口
) )
# 检查是否成功完成 # 检查是否成功完成
@ -57,8 +66,3 @@ class ODMProcessMonitor:
return True, "" return True, ""
return False, f"网格 {grid_idx + 1} 处理失败" return False, f"网格 {grid_idx + 1} 处理失败"
except Exception as e:
error_msg = f"执行异常: {str(e)}"
self.logger.error(error_msg)
return False, error_msg