Update quad_metric.py

Liyulingyue · May 25, 2024 · 72c2510 · 72c2510
1 parent afb949c
commit 72c2510
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/benchmark/PaddleOCR_DBNet/utils/ocr_metric/icdar2015/quad_metric.py b/benchmark/PaddleOCR_DBNet/utils/ocr_metric/icdar2015/quad_metric.py
@@ -61,7 +61,9 @@ def measure(self, batch, output, box_thresh=0.6):
  for i in range(pred_polygons.shape[0]):
  if pred_scores[i] >= box_thresh:
  # print(pred_polygons[i,:,:].tolist())
- pred.append(dict(points=pred_polygons[i, :, :].astype(np.int32)))
+ pred.append(
+ dict(points=pred_polygons[i, :, :].astype(np.int32))
+ )
  # pred = [dict(points=pred_polygons[i,:,:].tolist()) if pred_scores[i] >= box_thresh for i in range(pred_polygons.shape[0])]
  results.append(self.evaluator.evaluate_image(gt, pred))
  return results