将程序 stdout 写入文件

This commit is contained in:
William Jin 2024-09-26 17:07:06 +08:00
parent 31ba144974
commit 8420b3471a

View File

@ -177,15 +177,17 @@ def run_ocr(image):
"sudo", "-u", "tmfc",
"/home/tmfc/miniconda3/envs/got/bin/python3", "/home/tmfc/apps/got-ocr/GOT/demo/run_ocr_2.0_crop.py",
"--model-name", "/home/tmfc/apps/got-ocr/models/",
"--image-file", image_path,
"> /home/tmfc/apps/got-ocr/img.txt"
"--image-file", image_path
]
out_file = "/home/tmfc/apps/got-ocr/img.txt"
try:
subprocess.run(command, check=True)
with open(out_file, 'w') as f:
result = subprocess.run(command, stdout=f, stderr=subprocess.PIPE, text=True)
except subprocess.CalledProcessError as e:
return f"识别失败: {e}"
with open("/home/tmfc/apps/got-ocr/img.txt", 'r', encoding='utf-8') as f:
with open(out_file, 'r', encoding='utf-8') as f:
content = f.read()
return content