← Back to branch summary

~paparazzi-uav/paparazzi/v5.0-manual

« back to all changes in this revision

Viewing changes to sw/ext/opencv_bebop/opencv/modules/python/test/goodfeatures.py

  • Committer: Paparazzi buildbot
  • Date: 2016-05-18 15:00:29 UTC
  • Revision ID: felix.ruess+docbot@gmail.com-20160518150029-e8lgzi5kvb4p7un9
Manual import commit 4b8bbb730080dac23cf816b98908dacfabe2a8ec from v5.0 branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
sw/ext/opencv_bebop/opencv/modules/python/test/goodfeatures.py
 
1
#!/usr/bin/env python
 
2
 
 
3
import cv2.cv as cv
 
4
import unittest
 
5
 
 
6
class TestGoodFeaturesToTrack(unittest.TestCase):
 
7
    def test(self):
 
8
        arr = cv.LoadImage("../samples/c/lena.jpg", 0)
 
9
        original = cv.CloneImage(arr)
 
10
        size = cv.GetSize(arr)
 
11
        eig_image = cv.CreateImage(size, cv.IPL_DEPTH_32F, 1)
 
12
        temp_image = cv.CreateImage(size, cv.IPL_DEPTH_32F, 1)
 
13
        threshes = [ x / 100. for x in range(1,10) ]
 
14
 
 
15
        results = dict([(t, cv.GoodFeaturesToTrack(arr, eig_image, temp_image, 20000, t, 2, useHarris = 1)) for t in threshes])
 
16
 
 
17
        # Check that GoodFeaturesToTrack has not modified input image
 
18
        self.assert_(arr.tostring() == original.tostring())
 
19
 
 
20
        # Check for repeatability
 
21
        for i in range(10):
 
22
            results2 = dict([(t, cv.GoodFeaturesToTrack(arr, eig_image, temp_image, 20000, t, 2, useHarris = 1)) for t in threshes])
 
23
            self.assert_(results == results2)
 
24
 
 
25
        for t0,t1 in zip(threshes, threshes[1:]):
 
26
             r0 = results[t0]
 
27
             r1 = results[t1]
 
28
 
 
29
             # Increasing thresh should make result list shorter
 
30
             self.assert_(len(r0) > len(r1))
 
31
 
 
32
             # Increasing thresh should monly truncate result list
 
33
             self.assert_(r0[:len(r1)] == r1)
 
34
 
 
35
if __name__ == '__main__':
 
36
    unittest.main()