commit
5e9f48d384
3 changed files with 28 additions and 7 deletions
@ -0,0 +1,24 @@ |
||||
#!/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() |
Loading…
Reference in new issue