Explorar el Código

更新主函数默认参数,添加时间戳到输出文件名以避免覆盖

zhch158_admin hace 2 meses
padre
commit
f68e27e93e
Se han modificado 1 ficheros con 4 adiciones y 2 borrados
  1. 4 2
      compare_ocr_results.py

+ 4 - 2
compare_ocr_results.py

@@ -426,10 +426,12 @@ def main():
 if __name__ == "__main__":
     # 如果sys.argv没有被传入参数,则提供默认参数用于测试
     import sys
+    import time
     if len(sys.argv) == 1:
         sys.argv.extend([
-            './output/至远彩色印刷工业有限公司-2022年母公司_2.md', './sample_data/demo_54fa7ad0_page_1_nohf.md', 
-            '-o', './output/comparison_result', 
+            # './output/至远彩色印刷工业有限公司-2022年母公司_2.md', './sample_data/demo_54fa7ad0_page_1_nohf.md', 
+            './output/至远彩色印刷工业有限公司-2022年母公司_2.md', './output/至远彩色印刷工业有限公司-2022年母公司_2-GLM4.5V.md', 
+            '-o', f'./output/comparison_result_{time.strftime("%Y%m%d_%H%M%S")}', 
             '-f', 'both',
             '--ignore-images'])