2017-08-04 23:12:20 +08:00
|
|
|
#!/usr/bin/env python
|
|
|
|
|
|
|
|
'''
|
|
|
|
Algorithm serializaion test
|
|
|
|
'''
|
2017-12-11 17:55:03 +08:00
|
|
|
import cv2 as cv
|
2017-08-04 23:12:20 +08:00
|
|
|
|
|
|
|
from tests_common import NewOpenCVTests
|
|
|
|
|
|
|
|
class algorithm_rw_test(NewOpenCVTests):
|
|
|
|
def test_algorithm_rw(self):
|
|
|
|
# some arbitrary non-default parameters
|
2017-12-11 17:55:03 +08:00
|
|
|
gold = cv.AKAZE_create(descriptor_size=1, descriptor_channels=2, nOctaves=3, threshold=4.0)
|
|
|
|
gold.write(cv.FileStorage("params.yml", 1), "AKAZE")
|
2017-08-04 23:12:20 +08:00
|
|
|
|
2017-12-11 17:55:03 +08:00
|
|
|
fs = cv.FileStorage("params.yml", 0)
|
|
|
|
algorithm = cv.AKAZE_create()
|
2017-08-04 23:12:20 +08:00
|
|
|
algorithm.read(fs.getNode("AKAZE"))
|
|
|
|
|
|
|
|
self.assertEqual(algorithm.getDescriptorSize(), 1)
|
|
|
|
self.assertEqual(algorithm.getDescriptorChannels(), 2)
|
|
|
|
self.assertEqual(algorithm.getNOctaves(), 3)
|
|
|
|
self.assertEqual(algorithm.getThreshold(), 4.0)
|