Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
Tetragramm
GitHub Repository: Tetragramm/opencv
 Path: blob/master/modules/python/test/test_legacy.py
16337 views
1
#!/usr/bin/env python

2
from __future__ import print_function

3


4
import numpy as np

5
import cv2 as cv

6


7
from tests_common import NewOpenCVTests

8


9
class Hackathon244Tests(NewOpenCVTests):

10


11
    def test_int_array(self):

12
        a = np.array([-1, 2, -3, 4, -5])

13
        absa0 = np.abs(a)

14
        self.assertTrue(cv.norm(a, cv.NORM_L1) == 15)

15
        absa1 = cv.absdiff(a, 0)

16
        self.assertEqual(cv.norm(absa1, absa0, cv.NORM_INF), 0)

17


18
    def test_imencode(self):

19
        a = np.zeros((480, 640), dtype=np.uint8)

20
        flag, ajpg = cv.imencode("img_q90.jpg", a, [cv.IMWRITE_JPEG_QUALITY, 90])

21
        self.assertEqual(flag, True)

22
        self.assertEqual(ajpg.dtype, np.uint8)

23
        self.assertGreater(ajpg.shape[0], 1)

24
        self.assertEqual(ajpg.shape[1], 1)

25


26
    def test_projectPoints(self):

27
        objpt = np.float64([[1,2,3]])

28
        imgpt0, jac0 = cv.projectPoints(objpt, np.zeros(3), np.zeros(3), np.eye(3), np.float64([]))

29
        imgpt1, jac1 = cv.projectPoints(objpt, np.zeros(3), np.zeros(3), np.eye(3), None)

30
        self.assertEqual(imgpt0.shape, (objpt.shape[0], 1, 2))

31
        self.assertEqual(imgpt1.shape, imgpt0.shape)

32
        self.assertEqual(jac0.shape, jac1.shape)

33
        self.assertEqual(jac0.shape[0], 2*objpt.shape[0])

34


35
    def test_estimateAffine3D(self):

36
        pattern_size = (11, 8)

37
        pattern_points = np.zeros((np.prod(pattern_size), 3), np.float32)

38
        pattern_points[:,:2] = np.indices(pattern_size).T.reshape(-1, 2)

39
        pattern_points *= 10

40
        (retval, out, inliers) = cv.estimateAffine3D(pattern_points, pattern_points)

41
        self.assertEqual(retval, 1)

42
        if cv.norm(out[2,:]) < 1e-3:

43
            out[2,2]=1

44
        self.assertLess(cv.norm(out, np.float64([[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0]])), 1e-3)

45
        self.assertEqual(cv.countNonZero(inliers), pattern_size[0]*pattern_size[1])

46


47
    def test_fast(self):

48
        fd = cv.FastFeatureDetector_create(30, True)

49
        img = self.get_sample("samples/data/right02.jpg", 0)

50
        img = cv.medianBlur(img, 3)

51
        keypoints = fd.detect(img)

52
        self.assertTrue(600 <= len(keypoints) <= 700)

53
        for kpt in keypoints:

54
            self.assertNotEqual(kpt.response, 0)

55


56
    def check_close_angles(self, a, b, angle_delta):

57
        self.assertTrue(abs(a - b) <= angle_delta or

58
                        abs(360 - abs(a - b)) <= angle_delta)

59


60
    def check_close_pairs(self, a, b, delta):

61
        self.assertLessEqual(abs(a[0] - b[0]), delta)

62
        self.assertLessEqual(abs(a[1] - b[1]), delta)

63


64
    def check_close_boxes(self, a, b, delta, angle_delta):

65
        self.check_close_pairs(a[0], b[0], delta)

66
        self.check_close_pairs(a[1], b[1], delta)

67
        self.check_close_angles(a[2], b[2], angle_delta)

68


69
    def test_geometry(self):

70
        npt = 100

71
        np.random.seed(244)

72
        a = np.random.randn(npt,2).astype('float32')*50 + 150

73


74
        be = cv.fitEllipse(a)

75
        br = cv.minAreaRect(a)

76
        mc, mr = cv.minEnclosingCircle(a)

77


78
        be0 = ((150.2511749267578, 150.77322387695312), (158.024658203125, 197.57696533203125), 37.57804489135742)

79
        br0 = ((161.2974090576172, 154.41793823242188), (199.2301483154297, 207.7177734375), -9.164555549621582)

80
        mc0, mr0 = (160.41790771484375, 144.55152893066406), 136.713500977

81


82
        self.check_close_boxes(be, be0, 5, 15)

83
        self.check_close_boxes(br, br0, 5, 15)

84
        self.check_close_pairs(mc, mc0, 5)

85
        self.assertLessEqual(abs(mr - mr0), 5)

86


87


88
if __name__ == '__main__':

89
    NewOpenCVTests.bootstrap()

90


91