Ver código fonte

update confidence score 0.95->0.05

赵小蒙 1 ano atrás
pai
commit
3f062ad7ed
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      magic_pdf/model/magic_model.py

+ 1 - 1
magic_pdf/model/magic_model.py

@@ -48,7 +48,7 @@ class MagicModel:
             need_remove_list = []
             layout_dets = model_page_info["layout_dets"]
             for layout_det in layout_dets:
-                if layout_det["score"] < 0.95:
+                if layout_det["score"] <= 0.05:
                     need_remove_list.append(layout_det)
                 else:
                     continue