Переглянути джерело

更新测试模型函数的输入路径,移除默认值并使用新的示例图像

zhch158_admin 3 місяців тому
батько
коміт
38d3d068c6
1 змінених файлів з 3 додано та 2 видалено
  1. 3 2
      zhch/test_single_model.py

+ 3 - 2
zhch/test_single_model.py

@@ -41,7 +41,8 @@ MODEL_LIST = [
 
 def test_single_model(
     model_name: str, 
-    input_path: str = "sample_data/300674-母公司现金流量表-扫描.png", 
+    # input_path: str = "sample_data/300674-母公司现金流量表-扫描.png", 
+    input_path: str, 
     output_path: str = "./sample_data/output/"):
     """
     Test single model for layout detection.
@@ -88,6 +89,6 @@ if __name__ == "__main__":
         output_path.mkdir(parents=True, exist_ok=True)
 
         test_single_model(model_name=model_name, 
-                          input_path="sample_data/300674-母公司现金流量表-扫描.png", 
+                          input_path="sample_data/300674-合并损益变动表-扫描-1.png", 
                           output_path=output_path.as_posix())
         print("\n" + "="*50 + "\n")