|
|
|
@ -35,34 +35,5 @@ class aruco_test(NewOpenCVTests): |
|
|
|
|
img = cv.aruco.drawCharucoDiamond(aruco_dict, np.array([0, 1, 2, 3]), 100, 80) |
|
|
|
|
self.assertTrue(img is not None) |
|
|
|
|
|
|
|
|
|
def test_write_read_dict(self): |
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
aruco_dict = cv.aruco.getPredefinedDictionary(cv.aruco.DICT_5X5_50) |
|
|
|
|
markers_gold = aruco_dict.bytesList |
|
|
|
|
|
|
|
|
|
# write aruco_dict |
|
|
|
|
filename = "test_dict.yml" |
|
|
|
|
fs_write = cv.FileStorage(filename, cv.FileStorage_WRITE) |
|
|
|
|
aruco_dict.writeDictionary(fs_write) |
|
|
|
|
fs_write.release() |
|
|
|
|
|
|
|
|
|
# reset aruco_dict |
|
|
|
|
aruco_dict = cv.aruco.getPredefinedDictionary(cv.aruco.DICT_6X6_250) |
|
|
|
|
|
|
|
|
|
# read aruco_dict |
|
|
|
|
fs_read = cv.FileStorage(filename, cv.FileStorage_READ) |
|
|
|
|
aruco_dict.readDictionary(fs_read.root()) |
|
|
|
|
fs_read.release() |
|
|
|
|
|
|
|
|
|
# check equal |
|
|
|
|
self.assertEqual(aruco_dict.markerSize, 5) |
|
|
|
|
self.assertEqual(aruco_dict.maxCorrectionBits, 3) |
|
|
|
|
np.testing.assert_array_equal(aruco_dict.bytesList, markers_gold) |
|
|
|
|
|
|
|
|
|
finally: |
|
|
|
|
if os.path.exists(filename): |
|
|
|
|
os.remove(filename) |
|
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
|
|
NewOpenCVTests.bootstrap() |
|
|
|
|