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