subprocess.run
This commit is contained in:
parent
bb17302919
commit
6ad875d59a
@ -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,
|
||||||
|
@ -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
|
|
Loading…
Reference in New Issue
Block a user