diff --git a/open-anpr-core/src/test/java/com/visual/open/anpr/core/extract/PlateExtractorTest.java b/open-anpr-core/src/test/java/com/visual/open/anpr/core/extract/PlateExtractorTest.java index 4f8ed11..dcef774 100644 --- a/open-anpr-core/src/test/java/com/visual/open/anpr/core/extract/PlateExtractorTest.java +++ b/open-anpr-core/src/test/java/com/visual/open/anpr/core/extract/PlateExtractorTest.java @@ -27,7 +27,6 @@ public class PlateExtractorTest extends BaseTest { PlateExtractor extractor = new PlateExtractorImpl(torchPlateDetection, torchPlateRecognition); String imagePath = "open-anpr-core/src/test/resources/images"; -// String imagePath = "/Users/diven/workspace/pycharm/github/Chinese_license_plate_detection_recognition/imgs1/image001.jpg"; Map map = getImagePathMap(imagePath); for(String fileName : map.keySet()) { String imageFilePath = map.get(fileName); diff --git a/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateDetectionTest.java b/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateDetectionTest.java index 8c153c8..33d6a4f 100644 --- a/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateDetectionTest.java +++ b/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateDetectionTest.java @@ -11,8 +11,8 @@ import java.util.List; public class TorchPlateDetectionTest { public static void main(String[] args) { TorchPlateDetection torchPlateDetection = new TorchPlateDetection("open-anpr-core/src/main/resources/models/plate_detect.onnx", 1); -// String imagePath = "/Users/diven/workspace/idea/gitee/open-anpr/open-anpr-core/src/test/resources/images/image003.jpg"; - String imagePath = "/Users/diven/workspace/pycharm/github/Chinese_license_plate_detection_recognition/imgs3/double_yellow.jpg"; + + String imagePath = "open-anpr-core/src/test/resources/images"; ImageMat imageMat = ImageMat.fromImage(imagePath); List plateInfos = torchPlateDetection.inference(imageMat, 0.3f,0.5f, new HashMap<>()); System.out.println(plateInfos); diff --git a/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateRecognitionTest.java b/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateRecognitionTest.java index 440ac4a..979420c 100644 --- a/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateRecognitionTest.java +++ b/open-anpr-core/src/test/java/com/visual/open/anpr/core/models/TorchPlateRecognitionTest.java @@ -14,8 +14,7 @@ public class TorchPlateRecognitionTest { TorchPlateDetection torchPlateDetection = new TorchPlateDetection("open-anpr-core/src/main/resources/models/plate_detect.onnx", 1); TorchPlateRecognition torchPlateRecognition = new TorchPlateRecognition("open-anpr-core/src/main/resources/models/plate_rec_color.onnx", 1); -// String imagePath = "/Users/diven/workspace/pycharm/github/Chinese_license_plate_detection_recognition/imgs3/double_yellow.jpg"; - String imagePath = "/Users/diven/workspace/pycharm/github/Chinese_license_plate_detection_recognition/imgs2/image007.jpg"; + String imagePath = "open-anpr-core/src/test/resources/images"; ImageMat imageMat = ImageMat.fromImage(imagePath); List plateInfos = torchPlateDetection.inference(imageMat, 0.3f,0.5f, new HashMap<>()); System.out.println(plateInfos);