|
|
@@ -59,14 +59,25 @@ def main():
|
|
|
# 执行识别
|
|
|
result = pipeline.predict(input=args.input)
|
|
|
|
|
|
+ print(f"识别完成,结果类型: {type(result)}", file=sys.stderr)
|
|
|
+ print(f"结果数量: {len(result) if hasattr(result, '__len__') else 'N/A'}", file=sys.stderr)
|
|
|
+
|
|
|
# 保存结果
|
|
|
image_basename = os.path.splitext(os.path.basename(args.input))[0]
|
|
|
|
|
|
- for item in result:
|
|
|
+ for idx, item in enumerate(result):
|
|
|
+ print(f"处理结果项 {idx}: {type(item)}", file=sys.stderr)
|
|
|
+ print(f" - 有 save 方法: {hasattr(item, 'save')}", file=sys.stderr)
|
|
|
+
|
|
|
if hasattr(item, 'save'):
|
|
|
item_save_path = output_dir / image_basename
|
|
|
+ print(f" - 保存到: {item_save_path}", file=sys.stderr)
|
|
|
item.save(str(item_save_path))
|
|
|
- print(f"结果已保存到: {item_save_path}", file=sys.stderr)
|
|
|
+
|
|
|
+ # 列出保存后的文件
|
|
|
+ if item_save_path.exists():
|
|
|
+ files = list(item_save_path.iterdir())
|
|
|
+ print(f" - 保存后的文件列表: {[f.name for f in files]}", file=sys.stderr)
|
|
|
|
|
|
# 查找生成的 markdown 文件(多种可能的路径)
|
|
|
possible_paths = [
|
|
|
@@ -76,13 +87,15 @@ def main():
|
|
|
]
|
|
|
|
|
|
for markdown_file in possible_paths:
|
|
|
+ print(f" - 检查路径: {markdown_file} -> {markdown_file.exists()}", file=sys.stderr)
|
|
|
if markdown_file.exists():
|
|
|
print(f"SUCCESS: {markdown_file}")
|
|
|
sys.exit(0)
|
|
|
|
|
|
# 如果没找到,尝试递归查找所有 .md 文件
|
|
|
- import glob
|
|
|
- md_files = list(item_save_path.rglob("*.md"))
|
|
|
+ print(f" - 递归查找 .md 文件...", file=sys.stderr)
|
|
|
+ md_files = list(item_save_path.rglob("*.md")) if item_save_path.exists() else []
|
|
|
+ print(f" - 找到 {len(md_files)} 个 .md 文件", file=sys.stderr)
|
|
|
if md_files:
|
|
|
print(f"SUCCESS: {md_files[0]}")
|
|
|
sys.exit(0)
|