modules/java/android_test/src/org/opencv/test/features2d/FASTFeatureDetectorTest.java

   1 package org.opencv.test.features2d;
   2
   3 import java.util.Arrays;
   4
   5 import org.opencv.core.Core;
   6 import org.opencv.core.CvType;
   7 import org.opencv.core.Mat;
   8 import org.opencv.core.MatOfKeyPoint;
   9 import org.opencv.core.Point;
  10 import org.opencv.core.Scalar;
  11 import org.opencv.features2d.FeatureDetector;
  12 import org.opencv.core.KeyPoint;
  13 import org.opencv.test.OpenCVTestCase;
  14 import org.opencv.test.OpenCVTestRunner;
  15 import org.opencv.imgproc.Imgproc;
  16
  17 public class FASTFeatureDetectorTest extends OpenCVTestCase {
  18
  19     FeatureDetector detector;
  20     KeyPoint[] truth;
  21
  22     private Mat getMaskImg() {
  23         Mat mask = new Mat(100, 100, CvType.CV_8U, new Scalar(255));
  24         Mat right = mask.submat(0, 100, 50, 100);
  25         right.setTo(new Scalar(0));
  26         return mask;
  27     }
  28
  29     private Mat getTestImg() {
  30         Mat img = new Mat(100, 100, CvType.CV_8U, new Scalar(255));
  31         Imgproc.line(img, new Point(30, 30), new Point(70, 70), new Scalar(0), 8);
  32         return img;
  33     }
  34
  35     @Override
  36     protected void setUp() throws Exception {
  37         super.setUp();
  38         detector = FeatureDetector.create(FeatureDetector.FAST);
  39         truth = new KeyPoint[] { new KeyPoint(32, 27, 7, -1, 254, 0, -1), new KeyPoint(27, 32, 7, -1, 254, 0, -1), new KeyPoint(73, 68, 7, -1, 254, 0, -1),
  40                 new KeyPoint(68, 73, 7, -1, 254, 0, -1) };
  41     }
  42
  43     public void testCreate() {
  44         assertNotNull(detector);
  45     }
  46
  47     public void testDetectListOfMatListOfListOfKeyPoint() {
  48         fail("Not yet implemented");
  49     }
  50
  51     public void testDetectListOfMatListOfListOfKeyPointListOfMat() {
  52         fail("Not yet implemented");
  53     }
  54
  55     public void testDetectMatListOfKeyPoint() {
  56         Mat img = getTestImg();
  57         MatOfKeyPoint keypoints = new MatOfKeyPoint();
  58
  59         detector.detect(img, keypoints);
  60
  61         assertListKeyPointEquals(Arrays.asList(truth), keypoints.toList(), EPS);
  62
  63         // OpenCVTestRunner.Log("points found: " + keypoints.size());
  64         // for (KeyPoint kp : keypoints)
  65         // OpenCVTestRunner.Log(kp.toString());
  66     }
  67
  68     public void testDetectMatListOfKeyPointMat() {
  69         Mat img = getTestImg();
  70         Mat mask = getMaskImg();
  71         MatOfKeyPoint keypoints = new MatOfKeyPoint();
  72
  73         detector.detect(img, keypoints, mask);
  74
  75         assertListKeyPointEquals(Arrays.asList(truth[0], truth[1]), keypoints.toList(), EPS);
  76     }
  77
  78     public void testEmpty() {
  79         assertFalse(detector.empty());
  80     }
  81
  82     public void testRead() {
  83         String filename = OpenCVTestRunner.getTempFileName("yml");
  84
  85         writeFile(filename, "%YAML:1.0\nthreshold: 130\nnonmaxSuppression: 1\n");
  86         detector.read(filename);
  87
  88         MatOfKeyPoint keypoints1 = new MatOfKeyPoint();
  89
  90         detector.detect(grayChess, keypoints1);
  91
  92         writeFile(filename, "%YAML:1.0\nthreshold: 150\nnonmaxSuppression: 1\n");
  93         detector.read(filename);
  94
  95         MatOfKeyPoint keypoints2 = new MatOfKeyPoint();
  96
  97         detector.detect(grayChess, keypoints2);
  98
  99         assertTrue(keypoints2.total() <= keypoints1.total());
 100     }
 101
 102     public void testReadYml() {
 103         String filename = OpenCVTestRunner.getTempFileName("yml");
 104
 105         writeFile(filename,
 106                 "<?xml version=\"1.0\"?>\n<opencv_storage>\n<threshold>130</threshold>\n<nonmaxSuppression>1</nonmaxSuppression>\n</opencv_storage>\n");
 107         detector.read(filename);
 108
 109         MatOfKeyPoint keypoints1 = new MatOfKeyPoint();
 110
 111         detector.detect(grayChess, keypoints1);
 112
 113         writeFile(filename,
 114                 "<?xml version=\"1.0\"?>\n<opencv_storage>\n<threshold>150</threshold>\n<nonmaxSuppression>1</nonmaxSuppression>\n</opencv_storage>\n");
 115         detector.read(filename);
 116
 117         MatOfKeyPoint keypoints2 = new MatOfKeyPoint();
 118
 119         detector.detect(grayChess, keypoints2);
 120
 121         assertTrue(keypoints2.total() <= keypoints1.total());
 122     }
 123
 124     public void testWrite() {
 125         String filename = OpenCVTestRunner.getTempFileName("xml");
 126
 127         detector.write(filename);
 128
 129         String truth = "<?xml version=\"1.0\"?>\n<opencv_storage>\n<name>Feature2D.FAST</name>\n<nonmaxSuppression>1</nonmaxSuppression>\n<threshold>10</threshold>\n<type>2</type>\n</opencv_storage>\n";
 130         String data = readFile(filename);
 131         //Log.d("qqq", "\"" + data + "\"");
 132         assertEquals(truth, data);
 133     }
 134
 135     public void testWriteYml() {
 136         String filename = OpenCVTestRunner.getTempFileName("yml");
 137
 138         detector.write(filename);
 139
 140         String truth = "%YAML:1.0\nname: \"Feature2D.FAST\"\nnonmaxSuppression: 1\nthreshold: 10\ntype: 2\n";
 141         String data = readFile(filename);
 142
 143         //Log.d("qqq", "\"" + data + "\"");
 144         assertEquals(truth, data);
 145     }
 146 }