From bfb6ee9c19936b0b3cc1b5d80406af80583a057b Mon Sep 17 00:00:00 2001 From: diven Date: Tue, 17 Jan 2023 18:14:03 +0800 Subject: [PATCH] init --- .../open/anpr/core/extract/PlateExtractorImpl.java | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/open-anpr-core/src/main/java/com/visual/open/anpr/core/extract/PlateExtractorImpl.java b/open-anpr-core/src/main/java/com/visual/open/anpr/core/extract/PlateExtractorImpl.java index 6704a20..8c5259e 100644 --- a/open-anpr-core/src/main/java/com/visual/open/anpr/core/extract/PlateExtractorImpl.java +++ b/open-anpr-core/src/main/java/com/visual/open/anpr/core/extract/PlateExtractorImpl.java @@ -10,6 +10,7 @@ import com.visual.open.anpr.core.utils.CropUtil; import org.opencv.core.Mat; import java.util.HashMap; +import java.util.Iterator; import java.util.List; import java.util.Map; @@ -36,6 +37,14 @@ public class PlateExtractorImpl implements PlateExtractor { Mat crop = CropUtil.crop(image.toCvMat(), plateInfo.box); plateInfo.parseInfo = torchPlateRecognition.inference(ImageMat.fromCVMat(crop), plateInfo.single, new HashMap<>()); } + //清洗数据 + Iterator iterator = plateInfos.iterator(); + while (iterator.hasNext()) { + PlateInfo.ParseInfo parseInfo = iterator.next().parseInfo; + if(null == parseInfo || null == parseInfo.plateNo || parseInfo.plateNo.length() <= 4){ + iterator.remove(); + } + } //返回数据 return PlateImage.build(image.toBase64AndNoReleaseMat(), plateInfos); }