21 lines
405 B
Python
21 lines
405 B
Python
|
#!/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())
|