mirror of
https://github.com/opencv/opencv.git
synced 2024-12-14 08:59:11 +08:00
ef98c25d60
Features2d cleanup: Move several feature detectors and descriptors to opencv_contrib #25292 features2d cleanup: #24999 The PR moves KAZE, AKAZE, AgastFeatureDetector, BRISK and BOW to opencv_contrib/xfeatures2d. Related PR: opencv/opencv_contrib#3709
28 lines
942 B
Python
28 lines
942 B
Python
#!/usr/bin/env python
|
|
"""Algorithm serialization test."""
|
|
import tempfile
|
|
import os
|
|
import cv2 as cv
|
|
from tests_common import NewOpenCVTests
|
|
|
|
|
|
class algorithm_rw_test(NewOpenCVTests):
|
|
def test_algorithm_rw(self):
|
|
fd, fname = tempfile.mkstemp(prefix="opencv_python_algorithm_", suffix=".yml")
|
|
os.close(fd)
|
|
|
|
# some arbitrary non-default parameters
|
|
gold = cv.ORB_create(nfeatures=200, scaleFactor=1.3, nlevels=5, edgeThreshold=28)
|
|
gold.write(cv.FileStorage(fname, cv.FILE_STORAGE_WRITE), "ORB")
|
|
|
|
fs = cv.FileStorage(fname, cv.FILE_STORAGE_READ)
|
|
algorithm = cv.ORB_create()
|
|
algorithm.read(fs.getNode("ORB"))
|
|
|
|
self.assertEqual(algorithm.getMaxFeatures(), 200)
|
|
self.assertAlmostEqual(algorithm.getScaleFactor(), 1.3, places=6)
|
|
self.assertEqual(algorithm.getNLevels(), 5)
|
|
self.assertEqual(algorithm.getEdgeThreshold(), 28)
|
|
|
|
os.remove(fname)
|