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

update confidence score 0.95->0.05

赵小蒙 1 рік тому
батько
коміт
3f062ad7ed
1 змінених файлів з 1 додано та 1 видалено
  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