mirror of
https://github.com/opencv/opencv_contrib.git
synced 2025-10-24 20:01:12 +08:00
25 lines
858 B
Python
25 lines
858 B
Python
#!/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()
|