You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
858 B
24 lines
858 B
#!/usr/bin/env python |
|
import cv2 as cv |
|
import numpy as np |
|
|
|
from tests_common import NewOpenCVTests |
|
|
|
class disparity_test(NewOpenCVTests): |
|
def test_disp(self): |
|
# readGT |
|
ret,GT = cv.ximgproc.readGT(self.find_file("cv/disparityfilter/GT.png")) |
|
self.assertEqual(ret, 0) # returns 0 on success! |
|
self.assertFalse(np.shape(GT) == ()) |
|
|
|
# computeMSE |
|
left = cv.imread(self.find_file("cv/disparityfilter/disparity_left_raw.png"), cv.IMREAD_UNCHANGED) |
|
self.assertFalse(np.shape(left) == ()) |
|
left = np.asarray(left, dtype=np.int16) |
|
mse = cv.ximgproc.computeMSE(GT, left, (0, 0, GT.shape[1], GT.shape[0])) |
|
|
|
# computeBadPixelPercent |
|
bad = cv.ximgproc.computeBadPixelPercent(GT, left, (0, 0, GT.shape[1], GT.shape[0]), 24) |
|
|
|
if __name__ == '__main__': |
|
NewOpenCVTests.bootstrap()
|
|
|