Ver Fonte

fix(zhch): 修复单进程测试结果记录的路径错误

- 将 input_path.name 修改为 str(input_path),以记录完整的路径信息
- 更新错误处理和结果记录中的路径表示方式
- 注释掉测试模式的启动代码
zhch158_admin há 3 meses atrás
pai
commit
4872cb0d6c
1 ficheiros alterados com 4 adições e 4 exclusões
  1. 4 4
      zhch/ppstructurev3_single_process.py

+ 4 - 4
zhch/ppstructurev3_single_process.py

@@ -109,7 +109,7 @@ def process_images_single_process(image_paths: List[str],
                         
                         # 记录处理结果
                         batch_results.append({
-                            "image_path": input_path.name,
+                            "image_path": str(input_path),
                             "processing_time": batch_processing_time / len(batch),  # 平均时间
                             "success": True,
                             "device": device,
@@ -121,7 +121,7 @@ def process_images_single_process(image_paths: List[str],
                         print(f"Error saving result for {result.get('input_path', 'unknown')}: {e}", file=sys.stderr)
                         traceback.print_exc()
                         batch_results.append({
-                            "image_path": Path(result["input_path"]).name,
+                            "image_path": str(input_path),
                             "processing_time": 0,
                             "success": False,
                             "device": device,
@@ -151,7 +151,7 @@ def process_images_single_process(image_paths: List[str],
                 error_results = []
                 for img_path in batch:
                     error_results.append({
-                        "image_path": Path(img_path).name,
+                        "image_path": str(img_path),
                         "processing_time": 0,
                         "success": False,
                         "device": device,
@@ -320,6 +320,6 @@ if __name__ == "__main__":
             sys.argv.extend([f"--{key}", str(value)])
         
         # 测试模式
-        sys.argv.append("--test_mode")
+        # sys.argv.append("--test_mode")
     
     sys.exit(main())