TPS 薄板样条插值 python的opencv实现

原始论文:


Donato, G., & Belongie, S. J. (2003). Approximation methods for thin plate spline mappings and principal warps. Department of Computer Science and Engineering, University of California, San Diego.

效果:


原图:

变化后:

Code:


import cv2
import numpy as np
import random
# 首先读入img
img = cv2.imread('data/liv.png',cv2.IMREAD_COLOR)
img = cv2.resize(img,(180,32))
# N对基准控制点
N=5
points=[]
dx=int(180/(N-1))
for i in range(N):
    points.append((dx*i,4))
    points.append((dx*i,36))
# 周围拓宽一圈
img = cv2.copyMakeBorder(img,4,4,0,0,cv2.BORDER_REPLICATE)
# 画上绿色的圆圈
# for point in points:
# 	cv2.circle(img, point, 1, (0, 255, 0), 2)
tps = cv2.createThinPlateSplineShapeTransformer()

sourceshape = np.array(points,np.int32)
sourceshape=sourceshape.reshape(1,-1,2)
matches =[]
for i in range(1,N+1):
    matches.append(cv2.DMatch(i,i,0))

# 开始随机变动
newpoints=[]
PADDINGSIZ=10
for i in range(N):
    nx=points[i][0]+random.randint(0,PADDINGSIZ)-PADDINGSIZ/2
    ny=points[i][1]+random.randint(0,PADDINGSIZ)-PADDINGSIZ/2
    newpoints.append((nx,ny))
print(points,newpoints)
targetshape = np.array(newpoints,np.int32)
targetshape=targetshape.reshape(1,-1,2)
tps.estimateTransformation(sourceshape,targetshape ,matches)
tps.warpImage(img, img,cv2.INTER_CUBIC,cv2.BORDER_REPLICATE)
cv2.imwrite('tmp.png',img)

Debug参考:


https://xbuba.com/questions/41536344

https://docs.opencv.org/3.4/df/dfe/classcv_1_1ShapeTransformer.html

来源: https://blog.csdn.net/zhaominyiz/article/details/104114538