QRCodeDetectorTest.java 1.8 KB

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