瀏覽代碼

fix: handle empty input in prediction logic of predict_formula.py

myhloli 3 周之前
父節點
當前提交
b7a8347f45
共有 1 個文件被更改,包括 5 次插入2 次删除
  1. 5 2
      mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

+ 5 - 2
mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

@@ -129,8 +129,11 @@ class FormulaRecognizer(BaseOCRV20):
             new_idx: old_idx for new_idx, old_idx in enumerate(sorted_indices)
         }
 
-        # 进行预测
-        rec_formula = self.predict(sorted_images, batch_size)
+        if len(sorted_images) > 0:
+            # 进行预测
+            rec_formula = self.predict(sorted_images, batch_size)
+        else:
+            rec_formula = []
 
         # Restore original order
         unsorted_results = [""] * len(rec_formula)