diff --git a/ppocr/postprocess/picodet_postprocess.py b/ppocr/postprocess/picodet_postprocess.py index 9189c8ef5d2..1852e188091 100644 --- a/ppocr/postprocess/picodet_postprocess.py +++ b/ppocr/postprocess/picodet_postprocess.py @@ -292,6 +292,7 @@ def __call__(self, ori_img, img, preds): key=lambda x: x[1]["score"], reverse=True, )[0][0] - duplicate_idx.extend([x for x in overlaps if x != keep]) + if keep != i: + duplicate_idx.append(i) results = [x for i, x in enumerate(results) if i not in duplicate_idx] return results