Merge pull request #18018 from danielenricocahall:add-compose-panorama-python-binding

pull/18037/head
Alexander Alekhin 4 years ago
commit 3b337a12c9
  1. 2
      modules/stitching/include/opencv2/stitching.hpp
  2. 31
      modules/stitching/misc/python/test/test_stitching.py

@ -264,7 +264,7 @@ public:
@param pano Final pano. @param pano Final pano.
@return Status code. @return Status code.
*/ */
Status composePanorama(InputArrayOfArrays images, OutputArray pano); CV_WRAP Status composePanorama(InputArrayOfArrays images, OutputArray pano);
/** @overload */ /** @overload */
CV_WRAP Status stitch(InputArrayOfArrays images, OutputArray pano); CV_WRAP Status stitch(InputArrayOfArrays images, OutputArray pano);

@ -19,5 +19,36 @@ class stitching_test(NewOpenCVTests):
self.assertAlmostEqual(pano.shape[0], 685, delta=100, msg="rows: %r" % list(pano.shape)) self.assertAlmostEqual(pano.shape[0], 685, delta=100, msg="rows: %r" % list(pano.shape))
self.assertAlmostEqual(pano.shape[1], 1025, delta=100, msg="cols: %r" % list(pano.shape)) self.assertAlmostEqual(pano.shape[1], 1025, delta=100, msg="cols: %r" % list(pano.shape))
class stitching_compose_panorama_test_no_args(NewOpenCVTests):
def test_simple(self):
img1 = self.get_sample('stitching/a1.png')
img2 = self.get_sample('stitching/a2.png')
stitcher = cv.createStitcher(False)
stitcher.estimateTransform((img1, img2))
result, _ = stitcher.composePanorama()
assert result == 0
class stitching_compose_panorama_args(NewOpenCVTests):
def test_simple(self):
img1 = self.get_sample('stitching/a1.png')
img2 = self.get_sample('stitching/a2.png')
stitcher = cv.createStitcher(False)
stitcher.estimateTransform((img1, img2))
result, _ = stitcher.composePanorama((img1, img2))
assert result == 0
if __name__ == '__main__': if __name__ == '__main__':
NewOpenCVTests.bootstrap() NewOpenCVTests.bootstrap()

Loading…
Cancel
Save