Blame view

3rdparty/opencv-4.5.4/modules/objdetect/misc/java/test/QRCodeDetectorTest.java 1.76 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
  package org.opencv.test.objdetect;
  
  import java.util.List;
  import org.opencv.core.Mat;
  import org.opencv.objdetect.QRCodeDetector;
  import org.opencv.imgcodecs.Imgcodecs;
  import org.opencv.test.OpenCVTestCase;
  import java.util.Arrays;
  import java.util.ArrayList;
  import java.util.HashSet;
  import java.util.List;
  
  public class QRCodeDetectorTest extends OpenCVTestCase {
  
      private final static String ENV_OPENCV_TEST_DATA_PATH = "OPENCV_TEST_DATA_PATH";
      private String testDataPath;
  
      @Override
      protected void setUp() throws Exception {
          super.setUp();
  
          testDataPath = System.getenv(ENV_OPENCV_TEST_DATA_PATH);
          if (testDataPath == null)
              throw new Exception(ENV_OPENCV_TEST_DATA_PATH + " has to be defined!");
      }
  
      public void testDetectAndDecode() {
          Mat img = Imgcodecs.imread(testDataPath + "/cv/qrcode/link_ocv.jpg");
          assertFalse(img.empty());
          QRCodeDetector detector = new QRCodeDetector();
          assertNotNull(detector);
          String output = detector.detectAndDecode(img);
          assertEquals(output, "https://opencv.org/");
      }
  
      public void testDetectAndDecodeMulti() {
          Mat img = Imgcodecs.imread(testDataPath + "/cv/qrcode/multiple/6_qrcodes.png");
          assertFalse(img.empty());
          QRCodeDetector detector = new QRCodeDetector();
          assertNotNull(detector);
          List < String > output = new ArrayList< String >();
          boolean result = detector.detectAndDecodeMulti(img, output);
          assertTrue(result);
          assertEquals(output.size(), 6);
          List < String > expectedResults = Arrays.asList("SKIP", "EXTRA", "TWO STEPS FORWARD", "STEP BACK", "QUESTION", "STEP FORWARD");
          assertEquals(new HashSet<String>(output), new HashSet<String>(expectedResults));
      }
  }