Update quad_metric.py

Liyulingyue · May 25, 2024 · afb949c · afb949c
1 parent d3c643c
commit afb949c
Showing 1 changed file with 1 addition 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,7 @@ 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.int)))
+                        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