21 lines
405 B
Python
Executable File
21 lines
405 B
Python
Executable File
#!/usr/bin/python
|
|
|
|
import cv2 as cv
|
|
import numpy as np
|
|
import sys
|
|
|
|
img1 = cv.imread(sys.argv[1])
|
|
img1 = img1.astype(np.float32)
|
|
shift = np.array([5., 5.])
|
|
mapTest = cv.reg_MapShift(shift)
|
|
|
|
img2 = mapTest.warp(img1)
|
|
|
|
mapper = cv.reg_MapperGradShift()
|
|
mappPyr = cv.reg_MapperPyramid(mapper)
|
|
|
|
resMap = mappPyr.calculate(img1, img2)
|
|
mapShift = cv.reg.MapTypeCaster_toShift(resMap)
|
|
|
|
print(mapShift.getShift())
|