mirror of
https://github.com/opencv/opencv.git
synced 2026-07-21 19:33:03 +04:00
Lukas-Alexander Weber
d37bcbdc92
Introduce Cropping to OpenCV Stitching Tool * Introduced Cropping * integrate https://github.com/opencv/opencv/pull/21413/ * fixed failing subsetter tests * updated stuff
109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
import unittest
|
|
import os
|
|
import sys
|
|
|
|
import numpy as np
|
|
import cv2 as cv
|
|
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__),
|
|
'..', '..')))
|
|
|
|
from opencv_stitching.stitcher import Stitcher
|
|
# %%
|
|
|
|
|
|
class TestStitcher(unittest.TestCase):
|
|
|
|
def test_stitcher_aquaduct(self):
|
|
stitcher = Stitcher(nfeatures=250)
|
|
result = stitcher.stitch(["s1.jpg", "s2.jpg"])
|
|
cv.imwrite("result.jpg", result)
|
|
|
|
max_image_shape_derivation = 3
|
|
np.testing.assert_allclose(result.shape[:2],
|
|
(700, 1811),
|
|
atol=max_image_shape_derivation)
|
|
|
|
@unittest.skip("skip boat test (high resuolution ran >30s)")
|
|
def test_stitcher_boat1(self):
|
|
settings = {"warper_type": "fisheye",
|
|
"wave_correct_kind": "no",
|
|
"finder": "dp_colorgrad",
|
|
"compensator": "no",
|
|
"confidence_threshold": 0.3}
|
|
|
|
stitcher = Stitcher(**settings)
|
|
result = stitcher.stitch(["boat5.jpg", "boat2.jpg",
|
|
"boat3.jpg", "boat4.jpg",
|
|
"boat1.jpg", "boat6.jpg"])
|
|
|
|
cv.imwrite("boat_fisheye.jpg", result)
|
|
|
|
max_image_shape_derivation = 600
|
|
np.testing.assert_allclose(result.shape[:2],
|
|
(14488, 7556),
|
|
atol=max_image_shape_derivation)
|
|
|
|
@unittest.skip("skip boat test (high resuolution ran >30s)")
|
|
def test_stitcher_boat2(self):
|
|
settings = {"warper_type": "compressedPlaneA2B1",
|
|
"finder": "dp_colorgrad",
|
|
"compensator": "channel_blocks",
|
|
"confidence_threshold": 0.3}
|
|
|
|
stitcher = Stitcher(**settings)
|
|
result = stitcher.stitch(["boat5.jpg", "boat2.jpg",
|
|
"boat3.jpg", "boat4.jpg",
|
|
"boat1.jpg", "boat6.jpg"])
|
|
|
|
cv.imwrite("boat_plane.jpg", result)
|
|
|
|
max_image_shape_derivation = 600
|
|
np.testing.assert_allclose(result.shape[:2],
|
|
(7400, 12340),
|
|
atol=max_image_shape_derivation)
|
|
|
|
def test_stitcher_boat_aquaduct_subset(self):
|
|
settings = {"final_megapix": 1, "crop": True}
|
|
|
|
stitcher = Stitcher(**settings)
|
|
result = stitcher.stitch(["boat5.jpg",
|
|
"s1.jpg", "s2.jpg",
|
|
"boat2.jpg",
|
|
"boat3.jpg", "boat4.jpg",
|
|
"boat1.jpg", "boat6.jpg"])
|
|
cv.imwrite("subset_low_res.jpg", result)
|
|
|
|
max_image_shape_derivation = 100
|
|
np.testing.assert_allclose(result.shape[:2],
|
|
(705, 3374),
|
|
atol=max_image_shape_derivation)
|
|
|
|
def test_stitcher_budapest(self):
|
|
settings = {"matcher_type": "affine",
|
|
"estimator": "affine",
|
|
"adjuster": "affine",
|
|
"warper_type": "affine",
|
|
"wave_correct_kind": "no",
|
|
"confidence_threshold": 0.3}
|
|
|
|
stitcher = Stitcher(**settings)
|
|
result = stitcher.stitch(["budapest1.jpg", "budapest2.jpg",
|
|
"budapest3.jpg", "budapest4.jpg",
|
|
"budapest5.jpg", "budapest6.jpg"])
|
|
|
|
cv.imwrite("budapest.jpg", result)
|
|
|
|
max_image_shape_derivation = 50
|
|
np.testing.assert_allclose(result.shape[:2],
|
|
(1155, 2310),
|
|
atol=max_image_shape_derivation)
|
|
|
|
|
|
def starttest():
|
|
unittest.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
starttest()
|