mirror of
https://github.com/opencv/opencv_contrib.git
synced 2025-10-23 00:49:38 +08:00
shape: move sample to opencv_contrib
This commit is contained in:
26
modules/shape/test_python/test_shape.py
Normal file
26
modules/shape/test_python/test_shape.py
Normal file
@@ -0,0 +1,26 @@
|
||||
#!/usr/bin/env python
|
||||
import cv2 as cv
|
||||
|
||||
from tests_common import NewOpenCVTests
|
||||
|
||||
class shape_test(NewOpenCVTests):
|
||||
|
||||
def test_computeDistance(self):
|
||||
|
||||
a = self.get_sample('samples/data/shape_sample/1.png', cv.IMREAD_GRAYSCALE)
|
||||
b = self.get_sample('samples/data/shape_sample/2.png', cv.IMREAD_GRAYSCALE)
|
||||
|
||||
ca, _ = cv.findContours(a, cv.RETR_CCOMP, cv.CHAIN_APPROX_TC89_KCOS)
|
||||
cb, _ = cv.findContours(b, cv.RETR_CCOMP, cv.CHAIN_APPROX_TC89_KCOS)
|
||||
|
||||
hd = cv.createHausdorffDistanceExtractor()
|
||||
sd = cv.createShapeContextDistanceExtractor()
|
||||
|
||||
d1 = hd.computeDistance(ca[0], cb[0])
|
||||
d2 = sd.computeDistance(ca[0], cb[0])
|
||||
|
||||
self.assertAlmostEqual(d1, 26.4196891785, 3, "HausdorffDistanceExtractor")
|
||||
self.assertAlmostEqual(d2, 0.25804194808, 3, "ShapeContextDistanceExtractor")
|
||||
|
||||
if __name__ == '__main__':
|
||||
NewOpenCVTests.bootstrap()
|
Reference in New Issue
Block a user