Эх сурвалжийг харах

feat: 修改命令行参数,将dpi更改为pdf_dpi,以提高可读性

zhch158_admin 1 сар өмнө
parent
commit
a92c1e77f5

+ 3 - 3
zhch/mineru2_vllm_multthreads.py

@@ -455,7 +455,7 @@ def main():
     parser.add_argument("--server_url", type=str, default="http://127.0.0.1:8121", 
                        help="MinerU vLLM server URL")
     parser.add_argument("--timeout", type=int, default=300, help="Request timeout in seconds")
-    parser.add_argument("--dpi", type=int, default=200, help="PDF processing DPI")
+    parser.add_argument("--pdf_dpi", type=int, default=200, help="DPI for PDF to image conversion")
     parser.add_argument('--no-normalize', action='store_true', help='禁用数字标准化')
     parser.add_argument('--debug', action='store_true', help='启用调试模式')
     
@@ -572,7 +572,7 @@ def main():
             "server": args.server_url,
             "backend": "vlm-http-client",
             "timeout": args.timeout,
-            "pdf_dpi": args.dpi,
+            "pdf_dpi": args.pdf_dpi,
             "total_blocks": total_blocks,
             "block_type_stats": block_type_stats,
             "normalization_enabled": not args.no_normalize,
@@ -630,7 +630,7 @@ if __name__ == "__main__":
             "server_url": "http://10.192.72.11:8121",
             "timeout": "300",
             "batch_size": "1",
-            "dpi": "200",
+            "pdf_dpi": "200",
         }
         
         # 构造参数